Update app.py

app.py

@@ -2,8 +2,9 @@ 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_url, cached_download
 import joblib
+import requests
+from io import BytesIO
 
 app = FastAPI()
 
@@ -16,17 +17,19 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-REPO_ID = "GodfreyOwino/NPK_needs_mode2"  # Replace with your actual repo name
-FILENAME = "npk_needs_model.joblib"
+MODEL_URL = "https://huggingface.co/GodfreyOwino/NPK_needs_mode2/resolve/main/npk_needs_model.joblib"
 
 try:
-    model = joblib.load(cached_download(hf_hub_url(REPO_ID, FILENAME)))
+    response = requests.get(MODEL_URL)
+    response.raise_for_status()
+    model_bytes = BytesIO(response.content)
+    model = joblib.load(model_bytes)
     print("Model loaded successfully")
 except Exception as e:
     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...")
+    raise HTTPException(status_code=500, detail="Unable to load the model.")
 
 class InputData(BaseModel):
     features: list[float]