Filimize commited on
Commit
11c72ff
·
1 Parent(s): 05e3a24

Create new file

Browse files
Files changed (1) hide show
  1. app.py +39 -0
app.py ADDED
@@ -0,0 +1,39 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import joblib
2
+ import pandas as pd
3
+ import streamlit as st
4
+
5
+ model = joblib.load('model.joblib')
6
+ unique_values = joblib.load('unique_values.joblib')
7
+
8
+ unique_sex = unique_values['Sex']
9
+ unique_BP = unique_values['BP']
10
+ unique_Cholesterol = unique_values['Cholesterol']
11
+
12
+
13
+ def main():
14
+ st.title("Medicine Suggestion") #<-Apps' name
15
+
16
+ with st.form("questionaire"):
17
+ age = st.slider('Age',min_value=10,max_value=100)
18
+ Na_to_K = st.slider('Na_to_K',min_value=1,max_value=50)
19
+ sex = st.selectbox('sex',options=unique_sex)
20
+ BP = st.selectbox('BP',options=unique_BP)
21
+ Cholesterol = st.selectbox('Cholesterol',options=unique_Cholesterol)
22
+
23
+
24
+ # clicked==True only when the button is clicked
25
+ clicked = st.form_submit_button("Predict medicine")
26
+ if clicked:
27
+ result=model.predict(pd.DataFrame({"age": [age],
28
+ "Na_to_K": [Na_to_K],
29
+ "sex": [sex],
30
+ "BP": [BP],
31
+ "Cholesterol": [Cholesterol]
32
+ }))
33
+ # Show prediction
34
+ result = result[0]
35
+ st.success("You should get " +result)
36
+
37
+ # Run main()
38
+ if __name__=="__main__":
39
+ main()