Update app.py

app.py

@@ -22,20 +22,20 @@ def query(payload, api_url, token):
 
 # Define the function to run when the button is clicked
 def run_lora(prompt, selected_lora_index):
-    selected_lora = loras[selected_lora_index]
-    api_url = f"https://api-inference.huggingface.co/models/{selected_lora['repo']}"
-    trigger_word = selected_lora["trigger_word"]
-    token = os.getenv("API_TOKEN")
-    payload = {"inputs": f"{prompt} {trigger_word}"}
-    image_bytes = query(payload, api_url, token)
-    if image_bytes:
-        return Image.open(image_bytes)
-    else:
-        return "API Error"
+    if selected_lora_index is not None:  # Check if selected_lora_index is not None
+        selected_lora = loras[selected_lora_index]
+        api_url = f"https://api-inference.huggingface.co/models/{selected_lora['repo']}"
+        trigger_word = selected_lora["trigger_word"]
+        token = os.getenv("API_TOKEN")
+        payload = {"inputs": f"{prompt} {trigger_word}"}
+        image_bytes = query(payload, api_url, token)
+        if image_bytes:
+            return Image.open(image_bytes)
+    return "API Error or No LoRA selected"
 
 # Placeholder for gallery.select function
 def update_selection(selected):
-    return selected
+    return selected,  # Return as a tuple
 
 # Gradio UI
 with gr.Blocks(css="custom.css") as app:
@@ -52,7 +52,7 @@ with gr.Blocks(css="custom.css") as app:
             prompt_title = gr.Markdown("### Click on a LoRA in the gallery to select it")
             with gr.Row():
                 prompt = gr.Textbox(label="Prompt", show_label=False, lines=1, max_lines=1, placeholder="Type a prompt after selecting a LoRA")
-                button = gr.Button("Run")  # Added "Run" button
+                button = gr.Button("Run")
             result = gr.Image(interactive=False, label="Generated Image")
 
     gallery.select(