Spaces:
Sleeping
Sleeping
| from flask import Flask, request, jsonify | |
| import joblib | |
| import pandas as pd | |
| app = Flask(__name__) | |
| model = joblib.load("tuned_xgboost_model.pkl") | |
| def home(): | |
| return "SuperKart Sales Forecast Modal Deployment API" | |
| 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) |