Update app.py

app.py

@@ -60,13 +60,13 @@ def scrape_lora_link(url):
 
 def enable_lora(lora_scale,lora_add):
     if not lora_add:
-        gradio.Info("No Lora Loaded, Use basemodel", duration=5)
+        gr.Info("No Lora Loaded, Use basemodel", duration=5)
     else:
         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)
-        gradio.Info(f"{lora_add} Loaded", duration=5)
+        gr.Info(f"{lora_add} Loaded", duration=5)
 
 @spaces.GPU()
 def generate_image(
@@ -77,7 +77,8 @@ def generate_image(
     scales:float=3.5,
     steps:int=24,
     seed:int=-1,
-    nums:int=1):
+    nums:int=1,
+    progress=gr.Progress(track_tqdm=True)):
 
     pipe.to(device="cuda")