Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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('
|
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")
|