Update app.py

app.py

@@ -15,10 +15,9 @@ with open('loras.json', 'r') as f:
     loras = json.load(f)
 
 # Define the function to run when the button is clicked
-def update_selection(selected=None):
-    if selected is None:
-        return None
-    return {'index': selected[0]},
+def update_selection(selected_state: gr.SelectData):
+    logging.debug(f"Inside update_selection, selected_state: {selected_state}")
+    # Existing code...
 
 def run_lora(prompt, selected_state, progress=gr.Progress(track_tqdm=True)):
     logging.debug(f"Inside run_lora, selected_state: {selected_state}")