Spaces:
Build error
Build error
import joblib | |
import pandas as pd | |
import streamlit as st | |
model = joblib.load('model_tree.joblib') | |
def main(): | |
st.title("Network Ads") | |
with st.form("questionaire"): | |
age = st.slider('Age',min_value=10,max_value=100) | |
salary = st.slider('Estimated Salary',min_value=10000,max_value=100000) | |
clicked = st.form_submit_button("Predict income") | |
if clicked: | |
result=model.predict(pd.DataFrame({"age": [age],'Estimated Salary':[salary]})) | |
# Show prediction | |
result ='buy' if result[0]==1 else 'not buy' | |
st.success("You will " +result+" this product") | |
# Run main() | |
if __name__=="__main__": | |
main() |