Update app.py

app.py

@@ -10,12 +10,13 @@ Original file is located at
 import joblib
 import pandas as pd
 import streamlit as st
-model = joblib.load('model1.joblib')
+model = joblib.load('model.joblib')
 unique_values = joblib.load('unique.joblib')
 
 unique_LANDUSE_TYPE = unique_values["LANDUSE_TYPE"]
 unique_USER = unique_values["USER"]
 unique_Month = unique_values["Month"]
+unique_vendor = unique_values["VENDOR"]
 
 def main():
     st.title("Water usage in town predict")
@@ -23,12 +24,14 @@ def main():
         Land_use = st.selectbox("Land use",options = unique_LANDUSE_TYPE)
         User = st.selectbox("User",options = unique_USER)
         Month = st.selectbox("Month",options = unique_Month)
+        vendor = st.selectbox("Vendor",options = unique_vendor)
         pipe_diam = st.number_input("pipe diameter(meter):",min_value=0.5, max_value=10.0, step=0.5)
         clicked = st.form_submit_button("start predict ")
         if clicked:
             result=model.predict(pd.DataFrame({"LANDUSE_TYPE": [Land_use],
                                                "USER": [User],
                                                "Month": [Month],
+                                               "VENDOR": [vendor],
                                                "PIPE DIAM": [pipe_diam]}))
             str_result = str(result).strip("[]")
             st.success("Water usage predict value:"+str_result+"cubic meters")