Spaces:
Sleeping
Sleeping
File size: 677 Bytes
32e0955 03a8518 5a4ba5e aa7374b 03a8518 5a4ba5e 03a8518 5a4ba5e 32e0955 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 |
from flask import Flask, request, jsonify
import joblib
import pandas as pd
# Load trained model
model = joblib.load("best_model.pkl")
app = Flask(__name__)
@app.route("/")
def home():
return {"message": "SuperKart Model API is running!"}
@app.route("/predict", methods=["POST"])
def predict():
try:
data = request.get_json()
df = pd.DataFrame([data]) # Convert JSON to DataFrame
prediction = model.predict(df)[0]
return jsonify({"prediction": float(prediction)})
except Exception as e:
return jsonify({"error": str(e)})
if __name__ == "__main__":
app.run(host="0.0.0.0", port=7860) # Hugging Face expects 7860
|