Update app.py

app.py

@@ -362,7 +362,7 @@ def add_custom_lora(custom_lora, selected_indices):
                 if(repo.endswith(".safetensors") and repo.startswith("http")):
                     repo = download_file(repo)
                 new_item = {
-                    "image": image if image is not None else "",
+                    "image": image,
                     "title": title,
                     "repo": repo,
                     "weights": path,
@@ -390,22 +390,12 @@ def add_custom_lora(custom_lora, selected_indices):
             if len(selected_indices) >= 1:
                 lora1 = loras[selected_indices[0]]
                 selected_info_1 = f"### LoRA 1 Selected: {lora1['title']} ✨"
-                lora_image_1 = lora1['image']
+                lora_image_1 = lora1['image'] if lora1['image'] else None
             if len(selected_indices) >= 2:
                 lora2 = loras[selected_indices[1]]
                 selected_info_2 = f"### LoRA 2 Selected: {lora2['title']} ✨"
-                lora_image_2 = lora2['image']
+                lora_image_2 = lora2['image'] if lora2['image'] else None
             print("Chegou no final")
-            print (
-                gr.update(value=gallery_items),
-                selected_info_1,
-                selected_info_2,
-                selected_indices,
-                lora_scale_1,
-                lora_scale_2,
-                lora_image_1,
-                lora_image_2
-            )
             return (
                 gr.update(value=gallery_items),
                 selected_info_1,