Update app.py

app.py

@@ -62,9 +62,9 @@ def enable_lora(lora_scale, lora_in, lora_add):
     if not lora_in and not lora_add:
         return
     else:
-        if lora_in:
-            lora_add = lora_in
-        url = f'https://huggingface.co/{lora_add}/tree/main'
+        if lora_add:
+            lora_in = lora_add
+        url = f'https://huggingface.co/{lora_in}/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)   
@@ -76,13 +76,13 @@ def generate_image(
     scales:float=3.5,
     steps:int=24,
     seed:int=-1,
-    nums:int=1,
-    progress=gr.Progress(track_tqdm=True)):
+    nums:int=1):
 
     if seed == -1:
         seed = random.randint(0, MAX_SEED)
     seed = int(seed)
     print(f'prompt:{prompt}')
+    print(type(width))
     
     text = str(translator.translate(prompt, 'English'))
 
@@ -114,11 +114,13 @@ def gen(
     nums:int=1,
     lora_scale:float=1.0, 
     lora_in:str="", 
-    lora_add:str=""
+    lora_add:str="",
+    progress=gr.Progress(track_tqdm=True)
 ):
     enable_lora(lora_scale, lora_in, lora_add)
     pipe.to(device="cuda", dtype=torch.bfloat16)
-    generate_image(prompt,width,height,scales,steps,seed,nums)
+    return generate_image(prompt,width,height,scales,steps,seed,nums)
+     
 
 
 examples_bak = [
@@ -209,10 +211,9 @@ with gr.Blocks(css=CSS, js=JS, theme="Nymbo/Nymbo_Theme") as demo:
                     value=1.0,
                 )
                 lora_in = gr.Dropdown(
-                    choices=["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="",
                 )
                 lora_add = gr.Textbox(
                     label="Add Flux LoRA",