Upload app.py with huggingface_hub

app.py

@@ -5,6 +5,7 @@ import xgboost as xgb
 import numpy as np
 import pickle
 from huggingface_hub import hf_hub_download
+import os
 
 app = FastAPI(title="Headache Predictor API")
 
@@ -15,15 +16,22 @@ model = None
 async def load_model():
     global model
     try:
+        # Set cache directory to writable location
+        cache_dir = "/tmp/hf_cache"
+        os.makedirs(cache_dir, exist_ok=True)
+
         model_path = hf_hub_download(
             repo_id="emp-admin/headache-predictor-xgboost",
-            filename="model.pkl"
+            filename="model.pkl",
+            cache_dir=cache_dir
         )
         with open(model_path, 'rb') as f:
             model = pickle.load(f)
         print("✅ Model loaded successfully")
     except Exception as e:
         print(f"❌ Error loading model: {e}")
+        import traceback
+        traceback.print_exc()
 
 class PredictionRequest(BaseModel):
     features: list[float]