Rename utils.py to BiRefNetutils.py

2024-12-04 22:59:24 +07:00
parent f08eb04512
commit 29846c2ace
1 changed files with 0 additions and 0 deletions
@@ -0,0 +1,17 @@
+import os
+
+
+def check_download_model(model_path, repo_id="ViperYX/BiRefNet"):
+    if not os.path.exists(model_path):
+        folder_path = os.path.dirname(model_path)
+        if not os.path.exists(folder_path):
+            os.makedirs(folder_path)
+        file_name = os.path.basename(model_path)
+        print(f"Downloading BiRefNet model to: {model_path}")
+        from huggingface_hub import snapshot_download
+        snapshot_download(repo_id=repo_id,
+                        allow_patterns=[f"*{file_name}*"],
+                        local_dir=folder_path,
+                        local_dir_use_symlinks=False)
+        return True
+    return False