Update app.py

app.py

@@ -2,9 +2,8 @@ from fastapi import FastAPI, HTTPException
 from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel
 import numpy as np
-from huggingface_hub import hf_hub_download
+from huggingface_hub import hf_hub_url, cached_download
 import joblib
-import os
 
 app = FastAPI()
 
@@ -17,29 +16,17 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-REPO_ID = "GodfreyOwino/NPK_needs_mode2"
+REPO_ID = "GodfreyOwino/NPK_needs_mode2"  # Replace with your actual repo name
 FILENAME = "npk_needs_model.joblib"
 
-# Use a directory we're sure to have write permissions to
-MODEL_DIR = "/tmp/model"
-
-def get_model():
-    os.makedirs(MODEL_DIR, exist_ok=True)
-    try:
-        model_path = hf_hub_download(repo_id=REPO_ID, filename=FILENAME, cache_dir=MODEL_DIR)
-        return joblib.load(model_path)
-    except Exception as e:
-        print(f"Error downloading/loading model: {e}")
-        print(f"Error type: {type(e)}")
-        print(f"Error details: {str(e)}")
-        raise HTTPException(status_code=500, detail="Unable to download or find the model.")
-
 try:
-    model = get_model()
+    model = joblib.load(cached_download(hf_hub_url(REPO_ID, FILENAME)))
     print("Model loaded successfully")
 except Exception as e:
-    print(f"Error initializing model: {e}")
-    raise HTTPException(status_code=500, detail="Failed to initialize the model.")
+    print(f"Error loading model: {e}")
+    print(f"Error type: {type(e)}")
+    print(f"Error details: {str(e)}")
+    raise HTTPException(status_code=500, detail="Unable to download or find the model...")
 
 class InputData(BaseModel):
     features: list[float]