Woramaet's picture
Update app.py
52c80be
raw
history blame contribute delete
No virus
1.34 kB
import joblib
import pandas as pd
import streamlit as st
model = joblib.load('model.joblib')
unique_values = joblib.load('unique_values.joblib')
def main():
st.title("House price Paris")
with st.form("questionaire"):
numberOfRooms = st.slider("Number of Bedrooms", 0, 100, 1)
hasPool = st.slider("Has pool", 0, 1 ,1)
hasYard = st.slider("Has yard", 0, 1, 1)
squareMeters = st.slider("Square", 0, 100000, 1)
hasGuestRoom = st.slider("Number of Guest room", 0, 10, 1)
numPrevOwners = st.slider("Prev-Owners", 0, 10, 1)
# clicked==True only when the button is clicked
clicked = st.form_submit_button("Predict price")
if clicked:
result=model.predict(pd.DataFrame({"numberOfRooms": [numberOfRooms],
"hasPool": [hasPool],
"hasYard": [hasYard],
"squareMeters": [squareMeters],
"hasGuestRoom": [hasGuestRoom],
"numPrevOwners": [numPrevOwners]}))
# Show prediction
st.success(f"The house price prediction is {result}$")
if __name__ == "__main__":
main()