Upload folder using huggingface_hub

app.py

@@ -489,11 +489,16 @@ def push_to_hub(repo_name):
         return "No trained model found. Train first!"
     
     try:
-        from huggingface_hub import upload_file, create_repo
+        from huggingface_hub import upload_file, create_repo, login
+        
+        # Auth with secret
+        hf_token = os.environ.get("HF_TOKEN", "")
+        if hf_token:
+            login(token=hf_token)
         
         # Create model repo
         repo_id = repo_name if "/" in repo_name else f"COINjecture/{repo_name}"
-        create_repo(repo_id, repo_type="model", exist_ok=True)
+        create_repo(repo_id, repo_type="model", exist_ok=True, token=hf_token or None)
         
         # Upload model
         upload_file(
@@ -501,6 +506,7 @@ def push_to_hub(repo_name):
             path_in_repo="mu_net_trained.pt",
             repo_id=repo_id,
             repo_type="model",
+            token=hf_token or None,
         )
         
         # Create model card
@@ -542,6 +548,7 @@ license: mit
             path_in_repo="README.md",
             repo_id=repo_id,
             repo_type="model",
+            token=hf_token or None,
         )
         
         return f"✅ Model pushed to [{repo_id}](https://huggingface.co/{repo_id})"