Upload modutils.py

modutils.py

@@ -90,6 +90,7 @@ def get_private_model_list(repo_id, dir_path):
     try:
         files = api.list_repo_files(repo_id, token=hf_read_token)
     except Exception as e:
+        print(f"Error: Failed to list {repo_id}. ")
         return []
     model_list = []
     for file in files:
@@ -116,6 +117,9 @@ def get_private_lora_model_lists():
     return models
 
 
+private_lora_model_list = get_private_lora_model_lists()
+
+
 def download_private_file(repo_id, path, is_replace):
     from huggingface_hub import hf_hub_download
     file = Path(path)
@@ -126,6 +130,7 @@ def download_private_file(repo_id, path, is_replace):
     try:
         hf_hub_download(repo_id=repo_id, filename=filename, local_dir=dirname, use_auth_token=hf_read_token)
     except Exception as e:
+        print(f"Error: Failed to download {filename}. ")
         return
     if is_replace:
         file.resolve().rename(newpath.resolve())
@@ -229,10 +234,14 @@ def save_gallery_images(images, progress=gr.Progress(track_tqdm=True)):
         i += 1
         oldpath = Path(image[0])
         newpath = oldpath
-        if oldpath.exists():
-            newpath = oldpath.resolve().rename(Path(filename).resolve())
-        output_paths.append(str(newpath))
-        output_images.append((str(newpath), str(filename)))
+        try:
+            if oldpath.exists():
+                newpath = oldpath.resolve().rename(Path(filename).resolve())
+        except Exception:
+           pass
+        finally: 
+            output_paths.append(str(newpath))
+            output_images.append((str(newpath), str(filename)))
     progress(1, desc="Gallery updated.")
     return gr.update(value=output_images), gr.update(value=output_paths), gr.update(visible=True)