Spaces:
Runtime error
Runtime error
from flask import Flask, render_template, request | |
import pickle | |
import numpy as np | |
app = Flask(__name__) | |
# Load model and columns | |
model = pickle.load(open("car_price_model.pkl", "rb")) | |
columns = pickle.load(open("model_columns.pkl", "rb")) | |
# @app.route("/") | |
# def home(): | |
# return "Flask is working!" | |
def index(): | |
if request.method == "POST": | |
try: | |
present_price = float(request.form["present_price"]) | |
kms_driven = int(request.form["kms_driven"]) | |
owner = int(request.form["owner"]) | |
car_age = int(request.form["car_age"]) | |
fuel_type = request.form["fuel_type"] | |
company = request.form["company"] | |
# Prepare input dictionary | |
input_data = { | |
"kms_driven": kms_driven, | |
"Owner": owner, | |
"car_age": car_age, | |
"company_" + company: 1, | |
"fuel_type_" + fuel_type: 1 | |
} | |
input_vector = np.zeros(len(columns)) | |
for i, col in enumerate(columns): | |
if col in input_data: | |
input_vector[i] = input_data[col] | |
elif col == 'Present_Price': | |
input_vector[i] = present_price | |
predicted_price = model.predict([input_vector])[0] | |
return render_template("index.html", prediction_text=f"Estimated Selling Price: ₹ {predicted_price:,.2f}") | |
except Exception as e: | |
return render_template("index.html", prediction_text=f"Error: {str(e)}") | |
return render_template("index.html", prediction_text="") | |
if __name__ == "__main__": | |
app.run(debug=True,use_reloader=False) | |