Update app.py

app.py

@@ -303,12 +303,12 @@ def get_huggingface_safetensors(link):
                     image_url = f"https://huggingface.co/{link}/resolve/main/{image_elements[-1]}"
         except Exception as e:
             print(e)
-            raise Exception("Invalid Hugging Face repository with a *.safetensors LoRA")
+            raise gr.Error("Invalid Hugging Face repository with a *.safetensors LoRA")
         if not safetensors_name:
-            raise Exception("No *.safetensors file found in the repository")
+            raise gr.Error("No *.safetensors file found in the repository")
         return split_link[1], link, safetensors_name, trigger_word, image_url
     else:
-        raise Exception("Invalid Hugging Face repository link")
+        raise gr.Error("Invalid Hugging Face repository link")
 
 def check_custom_model(link):
     if link.endswith(".safetensors"):
@@ -338,7 +338,7 @@ def add_custom_lora(custom_lora, selected_indices):
             existing_item_index = next((index for (index, item) in enumerate(loras) if item['repo'] == repo), None)
             if existing_item_index is None:
                 new_item = {
-                    "image": image,
+                    "image": image if image is not None else "",
                     "title": title,
                     "repo": repo,
                     "weights": path,