Create app.py

app.py

@@ -0,0 +1,17 @@
+from fastapi import FastAPI
+import joblib
+import pandas as pd
+
+app = FastAPI()
+
+model = joblib.load('ModelV2.joblib')
+
+@app.post("/predict")
+async def predict(data: dict):
+    df = pd.DataFrame([data])
+    prediction = model.predict(df)
+    return {"prediction": prediction.tolist()}
+
+@app.get("/")
+async def root():
+    return {"message": "NPK Needs Prediction API"}