Update app.py

app.py

@@ -450,25 +450,21 @@ def get_civitai_safetensors(link):
   return model_data["name"], f"{lora_archive}/{safetensors_name}", trigger_word, image_url
   
 def check_custom_model(link):
-    try:
-        if(link.startswith("https://")):
-            if(link.startswith("https://huggingface.co") or link.startswith("https://www.huggingface.co")):
-                link_split = link.split("huggingface.co/")
-                return get_huggingface_safetensors(link_split[1])
-            elif(link.startswith("https://civitai.com") or link.startswith("https://www.civitai.com")):
-                return get_civitai_safetensors(link)
-        else: 
-            return get_huggingface_safetensors(link)
-    except Exception as e:
-        print("Error: ", e)
-        return None, None, None, None
+    if(link.startswith("https://")):
+        if(link.startswith("https://huggingface.co") or link.startswith("https://www.huggingface.co")):
+            link_split = link.split("huggingface.co/")
+            return get_huggingface_safetensors(link_split[1])
+        elif(link.startswith("https://civitai.com") or link.startswith("https://www.civitai.com")):
+            return get_civitai_safetensors(link)
+    else: 
+        return get_huggingface_safetensors(link)
 
 def show_loading_widget():
     return gr.update(visible=True)
 
 def load_custom_lora(link):
-    title, path, trigger_word, image = check_custom_model(link)
-    if(title):
+    try:
+        title, path, trigger_word, image = check_custom_model(link)
         card = f'''
         <div class="custom_lora_card">
           <span>Loaded custom LoRA:</span>
@@ -482,13 +478,10 @@ def load_custom_lora(link):
         </div>
         '''
         return gr.update(visible=True), card, gr.update(visible=True), [path, trigger_word], gr.Gallery(selected_index=None), f"Custom: {path}"
-    else:
+    except Exception as e:
         gr.Warning("Invalid LoRA: either you entered an invalid link, a non-SDXL LoRA or a LoRA with mature content")
         return gr.update(visible=True), "Invalid LoRA: either you entered an invalid link, a non-SDXL LoRA or a LoRA with mature content", gr.update(visible=False), None, gr.update(visible=True), gr.update(visible=True)
-    #except Exception as e:
-    #    gr.Info("Invalid custom LoRA")
-    #    return gr.update(visible=True), "Invalid LoRA: either you entered an invalid link, a non-SDXL LoRA or a LoRA with mature content", gr.update(visible=False), None, gr.update(visible=True), gr.update(visible=True)
-
+    
 def remove_custom_lora():
     return "", gr.update(visible=False), gr.update(visible=False), None
 with gr.Blocks(css="custom.css") as demo: