Puree commited on
Commit
d854e6e
1 Parent(s): 9b66b34

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -27,10 +27,10 @@ def main():
27
  # clicked==True only when the button is clicked
28
  clicked = st.form_submit_button("Predict income")
29
  if clicked:
30
- result=model.predict(pd.DataFrame({"Land_use": [Land_use],
31
- "User": [User],
32
- "Month": [Month],
33
- "pipe_diam": [pipe_diam]}))
34
  # Show prediction
35
  st.success("Water usage predict value:",result)
36
  if __name__ == '__main__':
 
27
  # clicked==True only when the button is clicked
28
  clicked = st.form_submit_button("Predict income")
29
  if clicked:
30
+ result=model.predict(pd.DataFrame({"LANDUSE_TYPE": [Land_use],
31
+ "USER": [User],
32
+ "MONTH": [Month],
33
+ "PIPE DIAM": [pipe_diam]}))
34
  # Show prediction
35
  st.success("Water usage predict value:",result)
36
  if __name__ == '__main__':