bug fix_3

app.py

@@ -63,25 +63,25 @@ if uploaded_file is not None:
        
         if predict_val == 0:
             probability = pred[0][predict_val]
-            rounded_probability = round(probability, 3)
+            rounded_probability = np.round(probability, 3)
             st.warning(" Stage: Mildly Demented") 
             st.warning(f" Prediction Probability: {rounded_probability}")
                      
         elif predict_val == 1:
             probability = pred[0][predict_val]
-            rounded_probability = round(probability, 3)
+            rounded_probability = np.round(probability, 3)
             st.warning(" Stage: Moderately Demented") 
             st.warning(f" Prediction Probability: {rounded_probability}")
                      
         elif predict_val == 2:
             probability = pred[0][predict_val]
-            rounded_probability = round(probability, 3)
+            rounded_probability = np.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]
-            rounded_probability = round(probability, 3)
+            rounded_probability = np.round(probability, 3)
             st.warning(" Stage: Very Mildly Demented") 
             st.warning(f" Prediction Probability: {rounded_probability}")