bug fix 2

app.py

@@ -63,19 +63,27 @@ if uploaded_file is not None:
        
         if predict_val == 0:
             probability = pred[0][predict_val]
-            st.warning(f"Mildly Demented with prediction probability of {probability}") 
+            rounded_probability = round(probability, 3)
+            st.warning(" Stage: Mildly Demented") 
+            st.warning(f" Prediction Probability: {rounded_probability}")
                      
         elif predict_val == 1:
             probability = pred[0][predict_val]
-            st.warning(f"Moderately Demented with prediction probability of {probability}") 
+            rounded_probability = round(probability, 3)
+            st.warning(" Stage: Moderately Demented") 
+            st.warning(f" Prediction Probability: {rounded_probability}")
                      
         elif predict_val == 2:
             probability = pred[0][predict_val]
-            st.warning(f"Not Demented with prediction probability of {probability}") 
+            rounded_probability = round(probability, 3)
+            st.warning(" Stage: Not Demented with prediction") 
+            st.warning(f" Prediction Probability: {rounded_probability}")
                     
         elif predict_val == 3:
             probability = pred[0][predict_val]
-            st.warning(f"Very Mildly Demented with prediction probability of {probability}") 
+            rounded_probability = round(probability, 3)
+            st.warning(" Stage: Very Mildly Demented") 
+            st.warning(f" Prediction Probability: {rounded_probability}")
                     
         else:
            st.warning("Error!")