Update app.py

app.py

@@ -304,6 +304,8 @@ def run_lora(prompt, image_input, image_strength, cfg_scale, steps, selected_ind
             yield image, seed, gr.update(value=progress_bar, visible=True)
         yield final_image, seed, gr.update(value=progress_bar, visible=False)
 
+run_lora.zerogpu = True
+
 def get_huggingface_safetensors(link):
     split_link = link.split("/")
     if len(split_link) == 2:
@@ -453,7 +455,7 @@ def remove_custom_lora(selected_indices):
         lora_image_2
     )
 
-run_lora.zerogpu = True
+
 
 css = '''
 #gen_btn{height: 100%}