Update app.py

app.py

@@ -59,13 +59,16 @@ def scrape_lora_link(url):
         return None
 
 def enable_lora(lora_scale, lora_in, lora_add):
-    if load_in:
-        lora_add = load_in
-    url = f'https://huggingface.co/{lora_add}/tree/main'
-    lora_name = scrape_lora_link(url)    
-    pipe.load_lora_weights(lora_add, weight_name=lora_name)
-    pipe.fuse_lora(lora_scale=lora_scale)
-    pipe.to(device="cuda", dtype=torch.bfloat16)
+    if not lora_in and not lora_add:
+        return
+    else:
+        if load_in:
+            lora_add = load_in
+        url = f'https://huggingface.co/{lora_add}/tree/main'
+        lora_name = scrape_lora_link(url)    
+        pipe.load_lora_weights(lora_add, weight_name=lora_name)
+        pipe.fuse_lora(lora_scale=lora_scale)
+        pipe.to(device="cuda", dtype=torch.bfloat16)
     
 @spaces.GPU()
 def generate_image(
@@ -207,7 +210,7 @@ with gr.Blocks(css=CSS, js=JS, theme="Nymbo/Nymbo_Theme") as demo:
                     value=1.0,
                 )
                 lora_in = gr.Dropdown(
-                    choice=["Shakker-Labs/FLUX.1-dev-LoRA-blended-realistic-illustration", "Shakker-Labs/AWPortrait-FL", ""], 
+                    choices=["Shakker-Labs/FLUX.1-dev-LoRA-blended-realistic-illustration", "Shakker-Labs/AWPortrait-FL", ""], 
                     label="LoRA Model", 
                     info="Load the LoRA",
                     value="Shakker-Labs/FLUX.1-dev-LoRA-blended-realistic-illustration",