Update app.py

app.py

@@ -106,16 +106,16 @@ def update_selection(evt: gr.SelectData, selected_indices, loras_state, width, h
     selected_info_2 = "Select a LoRA 2"
     lora_scale_1 = 1.15
     lora_scale_2 = 1.15
-    lora_image_1 = None
-    lora_image_2 = None
+    lora_image_1 = "/path/to/default/image.png"  # Provide a default image path
+    lora_image_2 = "/path/to/default/image.png"  # Provide a default image path
     if len(selected_indices) >= 1:
         lora1 = loras_state[selected_indices[0]]
         selected_info_1 = f"### LoRA 1 Selected: [{lora1['title']}](https://huggingface.co/{lora1['repo']}) ✨"
-        lora_image_1 = lora1['image']
+        lora_image_1 = lora1['image'] if lora1['image'] else lora_image_1  # Use default if None
     if len(selected_indices) >= 2:
         lora2 = loras_state[selected_indices[1]]
         selected_info_2 = f"### LoRA 2 Selected: [{lora2['title']}](https://huggingface.co/{lora2['repo']}) ✨"
-        lora_image_2 = lora2['image']
+        lora_image_2 = lora2['image'] if lora2['image'] else lora_image_2  # Use default if None
 
     if selected_indices:
         last_selected_lora = loras_state[selected_indices[-1]]