Update app.py

app.py

@@ -23,6 +23,7 @@ def query(payload, api_url, token):
 # Define the function to run when the button is clicked
 def run_lora(prompt, selected_lora_index):
     if selected_lora_index is not None:
+        selected_lora_index = selected_lora_index[0]  # Unpack the tuple
         selected_lora = loras[selected_lora_index]
         api_url = f"https://api-inference.huggingface.co/models/{selected_lora['repo']}"
         trigger_word = selected_lora["trigger_word"]
@@ -36,9 +37,8 @@ def run_lora(prompt, selected_lora_index):
             return Image.open(image_bytes)
     return "API Error or No LoRA selected"
 
-
 # Placeholder for gallery.select function
-def update_selection(selected):
+def update_selection(selected=None):
     if selected is None:
         return None,
     return selected[0],