Upload upload_to_hf.py

upload_to_hf.py

@@ -12,7 +12,8 @@ api = HfApi(token=HF_TOKEN)
 def get_files_to_upload():
     files = []
     ignore_dirs = {".git", "__pycache__", ".venv", "venv", "env", "node_modules", ".pytest_cache"}
-    ignore_files = {".env"}
+    ignore_files = {".env", "vortex.db"}
+    ignore_extensions = {".db", ".sqlite", ".sqlite3", ".pyc"}
     
     for root, dirs, files_list in os.walk("."):
         # Modify dirs in-place to exclude ignored directories
@@ -21,7 +22,7 @@ def get_files_to_upload():
                 dirs.remove(d)
         
         for f in files_list:
-            if f in ignore_files:
+            if f in ignore_files or any(f.endswith(ext) for ext in ignore_extensions):
                 continue
                 
             file_path = os.path.join(root, f)
@@ -49,14 +50,13 @@ def upload():
         print(f"[*] Uploading files...")
         for local_path, repo_path in files:
             print(f"    Uploading: {local_path} -> {repo_path}")
-            with open(local_path, 'rb') as f:
-                api.upload_file(
-                    path_or_fileobj=f,
-                    path_in_repo=repo_path,
-                    repo_id=REPO_ID,
-                    repo_type="space",
-                    commit_message=f"Upload {repo_path}",
-                )
+            api.upload_file(
+                path_or_fileobj=local_path,
+                path_in_repo=repo_path,
+                repo_id=REPO_ID,
+                repo_type="space",
+                commit_message=f"Upload {repo_path}",
+            )
         
         print("\n" + "="*50)
         print(f"[OK] SUCCESS!")