Upload folder using huggingface_hub

push_to_hf_space.py

@@ -1,26 +1,32 @@
-
 import os
 from huggingface_hub import HfApi
 
-DEPLOY_DIR = "deployment"
-
 def main():
     space_repo = os.environ.get("HF_SPACE_REPO")
+    hf_token = os.environ.get("HF_TOKEN")
+
     if not space_repo:
         raise ValueError("HF_SPACE_REPO environment variable is not set.")
+    if not hf_token:
+        raise ValueError("HF_TOKEN environment variable is not set.")
 
-    api = HfApi()
+    api = HfApi(token=hf_token)
+
+    # Create space if it doesn't exist
     api.create_repo(
         repo_id=space_repo,
         repo_type="space",
         space_sdk="docker",
         exist_ok=True
     )
+
+    # Upload deployment folder contents to Space
     api.upload_folder(
-        folder_path=DEPLOY_DIR,
+        folder_path="deployment",
         repo_id=space_repo,
         repo_type="space"
     )
+
     print("✅ Deployed to HF Space:", space_repo)
 
 if __name__ == "__main__":

requirements.txt

@@ -3,5 +3,6 @@ pandas==2.2.2
 numpy==2.0.2
 scikit-learn==1.6.1
 joblib==1.5.3
-huggingface_hub
 datasets
+huggingface_hub
+