prahalya's picture
Update app.py
4a97f35 verified
import streamlit as st
import pickle
import numpy as np
from sklearn.linear_model import LinearRegression
import pickle
import pandas as pd
# Load the trained model
with open("elite27.pkl", "rb") as f:
model = pickle.load(f)
st.title("🏑 House Price Prediction App")
# User input fields
square_feet = st.number_input("Enter Square Feet:", min_value=500, max_value=10000, )
bedrooms = st.number_input("Enter Number of Bedrooms:", min_value=1, max_value=10, )
bathrooms = st.number_input("Enter Number of Bathrooms:", min_value=1, max_value=10)
neighborhood = st.selectbox("Select Neighborhood --> 0:Rural 1:Semi Urban 2: Urban:", [0, 1, 2])
year_built = st.number_input("Enter Year Built:", min_value=1900, max_value=2025)
# Predict price
if st.button("Predict Price πŸ’°"):
user_data = np.array([[square_feet, bedrooms, bathrooms, neighborhood, year_built]], dtype=object)
prediction = model.predict(user_data)
st.success(f"🏠 Estimated House Price: ${prediction[0]:,.2f}")