Update app_flash.py

app_flash.py

@@ -70,35 +70,42 @@ from huggingface_hub import hf_hub_download, HfApi
 import os
 import tempfile
 from flashpack import FlashPackMixin
-from huggingface_hub import hf_hub_upload
+#from huggingface_hub import hf_hub_upload
+
+from huggingface_hub import upload_file
+import os
+import tempfile
+from flashpack import FlashPackMixin
+import torch
 
 def push_flashpack_to_hub_local(model: FlashPackMixin, hf_repo: str):
     logs = []
 
-    # Create a temporary directory for the pack
     with tempfile.TemporaryDirectory() as tmp_dir:
         logs.append(f"📂 Using temporary directory: {tmp_dir}")
 
-        # Ensure a proper file path (not just a directory)
+        # Save to a proper file path
         pack_file = os.path.join(tmp_dir, "model.pack")
         logs.append(f"💾 Saving FlashPack model to: {pack_file}")
 
-        # Save the model
         model.save_flashpack(pack_file, target_dtype=torch.float32)
         logs.append("✅ Model saved successfully.")
 
-        # Upload to HF Hub
+        # Upload to Hugging Face Hub
         logs.append(f"🌐 Uploading {pack_file} to Hugging Face repo: {hf_repo}")
-        hf_hub_upload(
-            repo_id=hf_repo,
-            path_or_fileobj=pack_file,
+        upload_file(
+            path_or_file=pack_file,
             path_in_repo="model.pack",  # filename in the repo
+            repo_id=hf_repo,
             repo_type="model",
-            token=os.environ.get("HF_TOKEN")  # make sure HF_TOKEN is set
+            token=os.environ.get("HF_TOKEN")  # ensure HF_TOKEN is set
         )
         logs.append("🚀 Upload successful!")
 
     return logs
+
+
+
 def train_and_push_flashpack(
     dataset_name: str = "gokaygokay/prompt-enhancer-dataset",
     hf_repo: str = "rahul7star/FlashPack",