Update app.py

app.py

@@ -1,31 +1,35 @@
 import os
+import requests
 from huggingface_hub import HfApi
 
-# Use Secrets (Settings > Variables and secrets > New secret)
-# Name the secret: HF_TOKEN
+# 1. Get Token from Secrets
 token = os.getenv("HF_TOKEN")
+dest_repo = "AIDev07/AIModelsLoaded"
+filename = "starcoder2-15b-instruct-v0.1-Q4_K_M.gguf"
+# Use the direct download URL
+source_url = f"https://huggingface.co{filename}"
 
 if not token:
-    print("❌ Error: HF_TOKEN secret not found in Space settings!")
+    print("❌ Error: Add your 'HF_TOKEN' to the Space Secrets!")
 else:
     api = HfApi(token=token)
-
-    # Configuration
-    source_url = "https://huggingface.co/nold/starcoder2-15b-GGUF/resolve/main/starcoder2-15b_Q4_K_M.gguf"
-    filename = "starcoder2-15b-instruct-v0.1-Q4_K_M.gguf"
-    dest_repo = "AIDev07/AIModelsLoaded"
-
-    print(f"🚀 Starting server-side transfer of {filename}...")
+    print(f"🚀 Starting Stream Transfer for: {filename}")
 
     try:
-        # This tells HF to grab the file from the URL and save it to your dataset
-        api.upload_file(
-            path_or_fileobj=source_url, # The URL acts as the source
-            path_in_repo=filename,
-            repo_id=dest_repo,
-            repo_type="dataset",
-            commit_message=f"Direct transfer of {filename}"
-        )
-        print("✅ Success! Check your dataset now.")
+        # 2. Open a streaming connection to the source file
+        with requests.get(source_url, stream=True) as r:
+            r.raise_for_status()
+            
+            # 3. Upload the stream directly to your dataset
+            # This doesn't save to your phone; it stays on HF servers.
+            api.upload_file(
+                path_or_fileobj=r.raw, 
+                path_in_repo=filename,
+                repo_id=dest_repo,
+                repo_type="dataset",
+                commit_message=f"Streamed upload of {filename}"
+            )
+            
+        print(f"✅ SUCCESS! {filename} is now in {dest_repo}")
     except Exception as e:
-        print(f"❌ Failed: {e}")
+        print(f"❌ Transfer Failed: {e}")