Spaces:
Runtime error
Runtime error
| # 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_predictor_api = Flask("Superkart Price Predictor") | |
| # Load the trained machine learning model | |
| model = joblib.load("deployment_files.ankur_supercart.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 Airbnb Rental Price Prediction API!" | |
| # Define an endpoint for single property prediction (POST request) | |
| def predict_superkart(): | |
| """ | |
| This function handles POST requests to the '/v1/superkart' 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 | |
| Superkart_data = request.get_json() | |
| # Extract relevant features from the JSON data | |
| sample = { | |
| 'Product_Id': Superkart_data['Product_Id'], | |
| 'Product_Weight': Superkart_data['Product_Weight'], | |
| 'Product_Sugar_Content': Superkart_data['Product_Sugar_Content'], | |
| 'Product_Allocated_Area': Superkart_data['Product_Allocated_Area'], | |
| 'Product_Type': Superkart_data['Product_Type'], | |
| 'Product_MRP': Superkart_data['Product_MRP'], | |
| 'Store_Id': Superkart_data['Store_Id'], | |
| 'Store_Establishment_Year': Superkart_data['Store_Size'], | |
| 'Store_Location_City_Type': Superkart_data['Store_Location_City_Type'], | |
| 'Store_Type': Superkart_data['Store_Type'], | |
| 'Product_Store_Sales_Total': Superkart_data['Product_Store_Sales_Total'], | |
| 'beds': Superkart_data['beds'] | |
| } | |
| # Convert the extracted data into a Pandas DataFrame | |
| input_data = pd.DataFrame([sample]) | |
| # Make prediction (get log_price) | |
| predicted_log_price = model.predict(input_data)[0] | |
| # Convert predicted_price to Python float | |
| predicted_price = round(float( predicted_log_price), 2) | |
| # 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 Price (in dollars)': predicted_log_price}) | |
| # Run the Flask application in debug mode if this script is executed directly | |
| if __name__ == '__main__': | |