Update app.py

app.py

@@ -374,6 +374,7 @@ def add_custom_lora(custom_lora, selected_indices, current_loras, gallery):
     else:
         return current_loras, gr.update(), "Select a LoRA 1", "Select a LoRA 2", "Select a LoRA 3", "Select a LoRA 4", "Select a LoRA 5", "Select a LoRA 6", selected_indices, 1.15, 1.15, 1.15, 1.15, 1.15, 1.15, None, None, None, None, None, None
 
+@spaces.GPU(duration=75)
 def generate_image(prompt_mash, steps, seed, cfg_scale, width, height, progress):
     print("Generating image...")
     pipe.to("cuda")
@@ -393,6 +394,7 @@ def generate_image(prompt_mash, steps, seed, cfg_scale, width, height, progress)
         ):
             yield img
 
+@spaces.GPU()
 def generate_image_to_image(prompt_mash, image_input_path, image_strength, steps, cfg_scale, width, height, seed):
     pipe_i2i.to("cuda")
     generator = torch.Generator(device="cuda").manual_seed(seed)
@@ -412,7 +414,7 @@ def generate_image_to_image(prompt_mash, image_input_path, image_strength, steps
     return final_image
 
 
-@spaces.GPU(duration=75)
+@spaces.GPU()
 def run_lora(prompt, image_input, image_strength, cfg_scale, steps, selected_indices, lora_scale_1, lora_scale_2, lora_scale_3, lora_scale_4, lora_scale_5, lora_scale_6, randomize_seed, seed, width, height, loras_state, progress=gr.Progress(track_tqdm=True)):
     if not selected_indices:
         raise gr.Error("You must select at least one LoRA before proceeding.")
@@ -715,7 +717,7 @@ with gr.Blocks(css=css, delete_cache=(60, 60)) as app:
         ])
     remove_lora_1event = remove_button_1.click(
         fn=remove_lora_1,
-        inputs=[0, selected_indices, loras_state],
+        inputs=[selected_indices, loras_state],
         outputs=[
             selected_info_1, selected_info_2, selected_info_3, selected_info_4, selected_info_5, selected_info_6,
             selected_indices,
@@ -725,7 +727,7 @@ with gr.Blocks(css=css, delete_cache=(60, 60)) as app:
     )
     remove_button_2.click(
         remove_lora_2,
-        inputs=[1, selected_indices, loras_state],
+        inputs=[selected_indices, loras_state],
         outputs=[
             selected_info_1, selected_info_2, selected_info_3, selected_info_4, selected_info_5, selected_info_6,
             selected_indices,
@@ -735,7 +737,7 @@ with gr.Blocks(css=css, delete_cache=(60, 60)) as app:
     )
     remove_button_3.click(
         remove_lora,
-        inputs=[2, selected_indices, loras_state],
+        inputs=[selected_indices, loras_state],
         outputs=[
             selected_info_1, selected_info_2, selected_info_3, selected_info_4, selected_info_5, selected_info_6,
             selected_indices,
@@ -745,7 +747,7 @@ with gr.Blocks(css=css, delete_cache=(60, 60)) as app:
     )
     remove_button_4.click(
         remove_lora,
-        inputs=[3, selected_indices, loras_state],
+        inputs=[selected_indices, loras_state],
         outputs=[
             selected_info_1, selected_info_2, selected_info_3, selected_info_4, selected_info_5, selected_info_6,
             selected_indices,
@@ -765,7 +767,7 @@ with gr.Blocks(css=css, delete_cache=(60, 60)) as app:
     )
     remove_button_6.click(
         remove_lora,
-        inputs=[5, selected_indices, loras_state],
+        inputs=[selected_indices, loras_state],
         outputs=[
             selected_info_1, selected_info_2, selected_info_3, selected_info_4, selected_info_5, selected_info_6,
             selected_indices,