Spaces:
Sleeping
Sleeping
| # Import necessary libraries | |
| import numpy as np | |
| import joblib # For loading the serialized model | |
| import pandas as pd # For data manipulation | |
| from flask import Flask, request, jsonify # For creating the Flask API | |
| # Initialize the Flask application | |
| superkart_sales_predictor_api = Flask("Superkart sales Predictor") | |
| # Load the trained machine learning model | |
| model = joblib.load("superkart_price_prediction_model_v1_0.joblib") | |
| # Define a route for the home page (GET request) | |
| 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!" | |
| # Define an endpoint for single sales prediction (POST request) | |
| def predict_sksales_price(): | |
| """ | |
| This function handles POST requests to the '/v1/sksales' endpoint. | |
| It expects a JSON payload containing property details and returns | |
| the predicted rental price as a JSON response. | |
| """ | |
| # Get the JSON data from the request body | |
| sksales_data = request.get_json() | |
| # Extract relevant features from the JSON data | |
| sample = { | |
| '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_Id': store_Id, | |
| 'store_Establishment_Year': store_Establishment_Year, | |
| 'store_Size': store_Size, | |
| 'store_Location_City_Type': store_Location_City_Type, | |
| 'store_Type': store_Type | |
| } | |
| # Convert the extracted data into a Pandas DataFrame | |
| input_data = pd.DataFrame([sample]) | |
| # Make prediction (get log_price) | |
| predicted_sales_price = model.predict(input_data)[0] | |
| # The conversion above is needed as we convert the model prediction (log price) to actual price using np.exp, which returns predictions as NumPy float32 values. | |
| # When we send this value directly within a JSON response, Flask's jsonify function encounters a datatype error | |
| # Return the actual price | |
| return jsonify({'Predicted sales price (in dollars)': predicted_sales_price}) | |
| # Run the Flask application in debug mode if this script is executed directly | |
| if __name__ == '__main__': | |
| superkart_sales_predictor_api.run(debug=True) | |