Update app.py

app.py

@@ -458,25 +458,28 @@ def show_loading_widget():
     return gr.update(visible=True)
 
 def load_custom_lora(link):
-    try:
-        title, path, trigger_word, image = check_custom_model(link)
-        card = f'''
-        <div class="custom_lora_card">
-          <span>Loaded custom LoRA:</span>
-          <div class="card_internal">
-            <img src="{image}" />
-            <div>
-                <h3>{title}</h3>
-                <small>{"Using: <code><b>"+trigger_word+"</code></b> as the trigger word" if trigger_word else "No trigger word found. If there's a trigger word, include it in your prompt"}<br></small>
+    if(link):
+        try:
+            title, path, trigger_word, image = check_custom_model(link)
+            card = f'''
+            <div class="custom_lora_card">
+              <span>Loaded custom LoRA:</span>
+              <div class="card_internal">
+                <img src="{image}" />
+                <div>
+                    <h3>{title}</h3>
+                    <small>{"Using: <code><b>"+trigger_word+"</code></b> as the trigger word" if trigger_word else "No trigger word found. If there's a trigger word, include it in your prompt"}<br></small>
+                </div>
+              </div>
             </div>
-          </div>
-        </div>
-        '''
-        return gr.update(visible=True), card, gr.update(visible=True), [path, trigger_word], gr.Gallery(selected_index=None), f"Custom: {path}"
-    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)
-    
+            '''
+            return gr.update(visible=True), card, gr.update(visible=True), [path, trigger_word], gr.Gallery(selected_index=None), f"Custom: {path}"
+        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)
+    else:
+        return gr.update(visible=False), "", 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: