Update prediction.py

prediction.py

@@ -11,6 +11,21 @@ from datetime import datetime
 from db import get_database  # Ensure this module is available
 import random
 
+from huggingface_hub import HfApi, HfFolder, upload_file
+
+# Login first (if not already)
+from huggingface_hub import notebook_login
+notebook_login()
+
+# Upload to your repo
+upload_file(
+    path_or_fileobj="transformer_model.bin",
+    path_in_repo="transformer_model.bin",
+    repo_id="your-username/your-space-repo",
+    repo_type="space"
+)
+
+
 # ------------------------ Ensuring Deterministic Behavior ------------------------
 random.seed(42)
 np.random.seed(42)