Update app.py

app.py

@@ -175,16 +175,18 @@ elif st.session_state.question_no == 10:
     test_data = np.array(transformed_data, dtype=np.float32)
     
     # Run the model
-    out = modelSession.run(None, {input_name: test_data})[1]
+    out = modelSession.run(None, {input_name: test_data})[0]
     # out = model.predict([st.session_state.pregnancies, st.session_state.glucose, st.session_state.bp, st.session_state.bmi, st.session_state.dpf])
-    out = softmax(list(out[0].values()))[1]
-    out = (out - 0.45)*50
-    out = 0 if out < 0 else 1 if out > 1 else out
-    risk = "low" if out <= 0.3 else "medium" if out <= 0.6 else "high" if out <= 0.9 else "very high"
-
-    st.write(f"Risk of type 2 diabetes is {risk}")
-
-    st.session_state.risk = risk
+    out = out[0]
+
+    if out == 0:
+        st.markdown("""
+            <b>CONGRATS</b> You are not in risk of diabetes type 2
+        """)
+    else:
+        st.markdown("""
+            You are in risk of type 2 diabetes. We recommend you to visit doctor as soon as possible.
+        """)
 
     if st.button("Retake test"):
         for key in st.session_state.keys():