Modal_deploy / app.py
Lokiiparihar's picture
Update app.py
52478dc verified
raw
history blame contribute delete
656 Bytes
from flask import Flask, request, jsonify
import joblib
import pandas as pd
app = Flask(__name__)
model = joblib.load("tuned_xgboost_model.pkl")
@app.route('/')
def home():
return "SuperKart Sales Forecast Modal Deployment API"
@app.route('/predict', methods=['POST'])
def predict():
try:
data = request.get_json()
input_df = pd.DataFrame([data])
prediction = model.predict(input_df)[0]
return jsonify({'Predicted_Sales': round(float(prediction), 2)})
except Exception as e:
return jsonify({'error': str(e)})
if __name__ == '__main__':
app.run(host='0.0.0.0', port=7860)