Upload app.py with huggingface_hub

app.py

@@ -1,42 +1,32 @@
-import os
+
 from flask import Flask, request, jsonify
 import joblib
 import pandas as pd
-from glob import glob
+import os
 
 app = Flask(__name__)
 
-# ---- Resolve model path robustly ----
-# Prefer env var, else default to the standard name we recommended.
-MODEL_PATH = os.getenv("MODEL_PATH", "superkart_sales_forecast.joblib")
-
-def load_model_bundle(path: str):
-    if not os.path.exists(path):
-        # Try to auto-find a .joblib in current dir (helpful during debugging)
-        candidates = glob("*.joblib")
-        raise FileNotFoundError(
-            f"Model file not found: {path}. "
-            f"Files present: {candidates}. "
-            f"Set MODEL_PATH env var to the correct file name."
-        )
-    bundle = joblib.load(path)
-    # Support both our recommended bundle {'pipeline': pipe} and plain pipeline
-    if isinstance(bundle, dict) and "pipeline" in bundle:
-        return bundle["pipeline"]
-    return bundle  # assume it's already a fitted pipeline
-
-pipe = load_model_bundle(MODEL_PATH)
-
-@app.get("/health")
+MODEL_PATH = os.getenv("MODEL_PATH", "best_model_random_forest.joblib")
+
+if not os.path.exists(MODEL_PATH):
+    raise FileNotFoundError(f"Model file not found: {MODEL_PATH}. Please ensure it's uploaded to the Space.")
+
+model_bundle = joblib.load(MODEL_PATH)
+if isinstance(model_bundle, dict) and "pipeline" in model_bundle:
+    pipe = model_bundle["pipeline"]
+else:
+    pipe = model_bundle
+
+@app.route("/health", methods=["GET"])
 def health():
-    return jsonify({"status": "ok", "model_path": MODEL_PATH})
+    return jsonify({"status": "ok", "model": MODEL_PATH})
 
-@app.post("/predict")
+@app.route("/predict", methods=["POST"])
 def predict():
     data = request.get_json(force=True)
     df = pd.DataFrame([data]) if isinstance(data, dict) else pd.DataFrame(data)
     preds = pipe.predict(df)
-    return jsonify({"predictions": [float(x) for x in preds]})
+    return jsonify({"predictions": [float(p) for p in preds]})
 
 if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=5000)
+    app.run(host="0.0.0.0", port=5000)