| | |
| | import numpy as np |
| | import joblib |
| | import pandas as pd |
| | from flask import Flask, request, jsonify |
| |
|
| | |
| | superkart_price_predictor_api = Flask("Superkart Sales Predictor") |
| |
|
| | |
| | model = joblib.load("superkart_price_prediction_model_v1_0.joblib") |
| |
|
| | |
| | @superkart_price_predictor_api.get('/') |
| | def home(): |
| | """ |
| | This function handles GET requests to the root URL ('/') of the API. |
| | It returns a simple welcome message. |
| | """ |
| | return "Welcome to the Superkart Sales Prediction API!" |
| |
|
| | |
| | @superkart_price_predictor_api.post('/v1/sales') |
| | def predict_sales_price(): |
| | """ |
| | This function handles POST requests to the '/v1/sales' endpoint. |
| | It expects a JSON payload containing property details and returns |
| | the predicted sales revenue price as a JSON response. |
| | """ |
| | |
| | property_data = request.get_json() |
| |
|
| | |
| |
|
| | input_data = pd.DataFrame([{ |
| | 'Product_Weight': Product_Weight, |
| | 'Product_Sugar_Content': Product_Sugar_Content, |
| | 'Product_Allocated_Area': Product_Allocated_Area, |
| | 'Product_Type': Product_Type, |
| | 'Product_MRP': Product_MRP, |
| | 'Store_Establishment_Year': Store_Establishment_Year, |
| | 'Store_Size': Store_Size, |
| | 'Store_Location_City_Type': Store_Location_City_Type, |
| | 'Store_Type': Store_Type |
| | }]) |
| |
|
| |
|
| | |
| | predicted_price = model.predict(input_data)[0] |
| |
|
| | |
| | return jsonify({'Predicted Price (in dollars)': predicted_price}) |
| |
|
| | |
| | if __name__ == '__main__': |
| | superkart_price_predictor_api.run(debug=True) |
| |
|