ve

src/app.py

@@ -44,11 +44,12 @@ os.environ['HF_HOME'] = str(cache_dir)
 # Load model from Hugging Face Hub
 def load_model():
     try:
-        # Download the model file from HF Hub with temp directory
+        # Download the model file from HF Hub with authentication
         model_path = hf_hub_download(
             repo_id=MODEL_ID,
             filename="best_model.pth",
-            cache_dir=cache_dir
+            cache_dir=cache_dir,
+            token=os.environ.get('HF_TOKEN')  # Get token from environment variable
         )
         
         # Initialize our custom GPT model