Puree commited on
Commit
30ac382
1 Parent(s): 8ba8a2d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -10,12 +10,13 @@ Original file is located at
10
  import joblib
11
  import pandas as pd
12
  import streamlit as st
13
- model = joblib.load('model1.joblib')
14
  unique_values = joblib.load('unique.joblib')
15
 
16
  unique_LANDUSE_TYPE = unique_values["LANDUSE_TYPE"]
17
  unique_USER = unique_values["USER"]
18
  unique_Month = unique_values["Month"]
 
19
 
20
  def main():
21
  st.title("Water usage in town predict")
@@ -23,12 +24,14 @@ def main():
23
  Land_use = st.selectbox("Land use",options = unique_LANDUSE_TYPE)
24
  User = st.selectbox("User",options = unique_USER)
25
  Month = st.selectbox("Month",options = unique_Month)
 
26
  pipe_diam = st.number_input("pipe diameter(meter):",min_value=0.5, max_value=10.0, step=0.5)
27
  clicked = st.form_submit_button("start predict ")
28
  if clicked:
29
  result=model.predict(pd.DataFrame({"LANDUSE_TYPE": [Land_use],
30
  "USER": [User],
31
  "Month": [Month],
 
32
  "PIPE DIAM": [pipe_diam]}))
33
  str_result = str(result).strip("[]")
34
  st.success("Water usage predict value:"+str_result+"cubic meters")
 
10
  import joblib
11
  import pandas as pd
12
  import streamlit as st
13
+ model = joblib.load('model.joblib')
14
  unique_values = joblib.load('unique.joblib')
15
 
16
  unique_LANDUSE_TYPE = unique_values["LANDUSE_TYPE"]
17
  unique_USER = unique_values["USER"]
18
  unique_Month = unique_values["Month"]
19
+ unique_vendor = unique_values["VENDOR"]
20
 
21
  def main():
22
  st.title("Water usage in town predict")
 
24
  Land_use = st.selectbox("Land use",options = unique_LANDUSE_TYPE)
25
  User = st.selectbox("User",options = unique_USER)
26
  Month = st.selectbox("Month",options = unique_Month)
27
+ vendor = st.selectbox("Vendor",options = unique_vendor)
28
  pipe_diam = st.number_input("pipe diameter(meter):",min_value=0.5, max_value=10.0, step=0.5)
29
  clicked = st.form_submit_button("start predict ")
30
  if clicked:
31
  result=model.predict(pd.DataFrame({"LANDUSE_TYPE": [Land_use],
32
  "USER": [User],
33
  "Month": [Month],
34
+ "VENDOR": [vendor],
35
  "PIPE DIAM": [pipe_diam]}))
36
  str_result = str(result).strip("[]")
37
  st.success("Water usage predict value:"+str_result+"cubic meters")