Add Progressbar

app.py

@@ -24,7 +24,7 @@ def download_file(url, filename, progress=gr.Progress(track_tqdm=True)):
     if total_size_in_bytes != 0 and progress_bar.n != total_size_in_bytes:
         print("ERROR, something went wrong")
 
-def get_civitai_model_info(model_id):
+def get_civitai_model_info(model_id, progress=gr.Progress(track_tqdm=True)):
     url = f"https://civitai.com/api/v1/models/{model_id}"
     response = requests.get(url)
     if response.status_code != 200:
@@ -37,7 +37,7 @@ def find_download_url(data, file_extension):
             return file['downloadUrl']
     return None
 
-def download_civitai_model(model_id, lora_id=""):
+def download_civitai_model(model_id, lora_id="", progress=gr.Progress(track_tqdm=True)):
     try:
         model_data = get_civitai_model_info(model_id)
         if model_data is None: