Divar / divar.py
radi02's picture
Upload 4 files
b51ee04
raw
history blame
677 Bytes
import streamlit as st
import numpy as np
import pandas as pd
import joblib
model = joblib.load('divar.sav')
st.title('what is the price of your home??')
Area = st.number_input("Input Area", 0,10000)
Room = st.selectbox("Room? ", [1,2,3,4,5,6,7,8,9,10])
Parking = st.selectbox("Parking??", ['1','0])
Warehouse = st.selectbox("Parking??", ['1','0])
Elevator = st.selectbox("Parking??", ['1','0])
Adress = st.text_input("Adress", 'Abazar')
def predict():
row = np.array([Area,Room,Parking,Warehouse,Elevator,Adress])
X = pd.DataFrame([row], columns = columns)
prediction = model.predict(X)
print(prediction)
trigger = st.button('Predict', on_click=predict)