Update app.py

app.py

@@ -1,4 +1,17 @@
+import json
 import streamlit as st
+from sklearn.preprocessing import StandardScaler
+from sklearn.ensemble import RandomForestClassifier
+
+with open("random_forest_model.json", "r") as f:
+    params = json.load(f)
+    model = RandomForestClassifier(**params)
+    del(params)
+
+with open("scaler.json", "r") as f:
+    params = json.load(f)
+    scaler = StandardScaler(**params)
+    del(params)
 
 def calculate_bmi(weight, height):
     # Calculate BMI
@@ -123,7 +136,9 @@ elif st.session_state.question_no == 9:
         st.session_state.dpf = diabetic_pedigree_function(st.session_state.diabeticMother, st.session_state.diabeticFather, st.session_state.diabeticSibling, st.session_state.diabeticSibling)
         st.session_state.question_no += 1
 
-#elif st.session_state.question_no == 10:
+elif st.session_state.question_no == 10:
+    out = model.predict([st.session_state.pregnancies, st.session_state.glucose, st.session.state.bp, st.bmi, st.dpf])
+    st.write(out)