Update app.py

app.py

@@ -706,7 +706,7 @@ with gr.Blocks(css=css, delete_cache=(60, 60)) as app:
 
     gallery.select(
         fn=update_selection,
-        inputs=[selected_indices, loras_state, width, height],
+        inputs=[0, selected_indices, loras_state, width, height],
         outputs=[
             prompt,
             selected_info_1, selected_info_2, selected_info_3, selected_info_4, selected_info_5, selected_info_6,
@@ -736,7 +736,7 @@ with gr.Blocks(css=css, delete_cache=(60, 60)) as app:
         ]
     )
     remove_button_3.click(
-        remove_lora,
+        remove_lora_3,
         inputs=[selected_indices, loras_state],
         outputs=[
             selected_info_1, selected_info_2, selected_info_3, selected_info_4, selected_info_5, selected_info_6,
@@ -746,7 +746,7 @@ with gr.Blocks(css=css, delete_cache=(60, 60)) as app:
         ]
     )
     remove_button_4.click(
-        remove_lora,
+        remove_lora_4,
         inputs=[selected_indices, loras_state],
         outputs=[
             selected_info_1, selected_info_2, selected_info_3, selected_info_4, selected_info_5, selected_info_6,
@@ -756,8 +756,8 @@ with gr.Blocks(css=css, delete_cache=(60, 60)) as app:
         ]
     )
     remove_button_5.click(
-        remove_lora,
-        inputs=[4, selected_indices, loras_state],
+        remove_lora_5,
+        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,
@@ -766,7 +766,7 @@ with gr.Blocks(css=css, delete_cache=(60, 60)) as app:
         ]
     )
     remove_button_6.click(
-        remove_lora,
+        remove_lora_6,
         inputs=[selected_indices, loras_state],
         outputs=[
             selected_info_1, selected_info_2, selected_info_3, selected_info_4, selected_info_5, selected_info_6,