Upload app.py

app.py

@@ -69,31 +69,29 @@ def main():
         Heart_Rate = st.slider("Heart Rate", min_value=65, max_value=86)
         Daily_Steps = st.slider("Daily Steps", min_value=3000, max_value=10000)
 
-        clicked = st.form_submit_button("Predict Sleep Health")
-        if clicked:
-            result=model.predict(pd.DataFrame({"Gender": [Gender_DICT[Gender]],
-                                               "Age": [age],
-                                               "Occupation": [Occupation_DICT[occupation]],
-                                               "Sleep Duration": [Sleep_Duration],
-                                               "Quality of Sleep": [Quality_of_Sleep],
-                                               "Physical Activity Level": [Physical_Activity_Level],
-                                               "Stress Level":[Stress_Level],
-                                               "BMI Category": [BMI_DICT[BMI_Category]],
-                                               "Blood Pressure": [Blood_DICT[Blood_Pressure]],
-                                               "Heart Rate": [Heart_Rate],
-                                               "Daily Steps": [Daily_Steps]
-                                               }))
-            clicked = st.form_submit_button("Predict Sleep Health")
-            if clicked:
-                if result[0] == 1:
-                    result_message = "None"
-                elif result[0] == 2:
-                    result_message = "Sleep Apnea"
-                elif result[0] == 3:
-                    result_message = "Insomnia"
-                else:
-                    result_message = "Unknown"  # กำหนดค่าเริ่มต้นให้กับ result_message ในกรณีอื่นๆ
-                st.success(result_message)
+        clicked_sleep = st.form_submit_button("Predict Sleep Health")
+        if clicked_sleep:
+            result = model.predict(pd.DataFrame({"Gender": [Gender_DICT[Gender]],
+                                          "Age": [age],
+                                          "Occupation": [Occupation_DICT[occupation]],
+                                          "Sleep Duration": [Sleep_Duration],
+                                          "Quality of Sleep": [Quality_of_Sleep],
+                                          "Physical Activity Level": [Physical_Activity_Level],
+                                          "Stress Level": [Stress_Level],
+                                          "BMI Category": [BMI_DICT[BMI_Category]],
+                                          "Blood Pressure": [Blood_DICT[Blood_Pressure]],
+                                          "Heart Rate": [Heart_Rate],
+                                          "Daily Steps": [Daily_Steps]
+                                          }))
+            if result[0] == 1:
+                result_message = "None"
+            elif result[0] == 2:
+                result_message = "Sleep Apnea"
+            elif result[0] == 3:
+                result_message = "Insomnia"
+            else:
+                result_message = "Unknown"  # กำหนดค่าเริ่มต้นให้กับ result_message ในกรณีอื่นๆ
+            st.success(result_message)
 
 if __name__=='__main__':
     main()