Update app.py

app.py

@@ -13,12 +13,16 @@ with open('loras.json', 'r') as f:
     loras = json.load(f)
 
 def update_selection(selected_state):
-    selected_lora_index = selected_state.index
-    selected_lora = loras[selected_lora_index]
-    new_placeholder = f"Type a prompt for {selected_lora['title']}"
-    lora_repo = selected_lora["repo"]
-    updated_text = f"### Selected: [{lora_repo}](https://huggingface.co/{lora_repo}) ✨"
-    return (gr.update(placeholder=new_placeholder), updated_text, selected_state)
+    if selected_state is not None:
+        selected_lora_index = selected_state.index
+        selected_lora = loras[selected_lora_index]
+        new_placeholder = f"Type a prompt for {selected_lora['title']}"
+        lora_repo = selected_lora["repo"]
+        updated_text = f"### Selected: [{lora_repo}](https://huggingface.co/{lora_repo}) ✨"
+        return (gr.update(placeholder=new_placeholder), updated_text, selected_state)
+    else:
+        return (gr.update(placeholder="Select a Lora first"), "No Lora selected", None)
+
 
 def run_lora(prompt, selected_state, progress=gr.Progress(track_tqdm=True)):
     selected_lora_index = selected_state.index