updates

app.py

@@ -62,28 +62,24 @@ if uploaded_file is not None:
         predict_val = np.argmax(pred, axis=1)
        
         if predict_val == 0:
-            probability = pred[0][predict_val]
-            rounded_probability = "{:.3f}".format(probability)
+            probability = pred[0][predict_val][0]            
             st.markdown(" Stage: Mildly Demented") 
-            st.markdown(f" Prediction Probability: {rounded_probability}")
+            st.markdown(f" Prediction Probability: {probability}")
                      
         elif predict_val == 1:
-            probability = pred[0][predict_val]
-            rounded_probability = "{:.3f}".format(probability)
+            probability = pred[0][predict_val][0]           
             st.markdown(" Stage: Moderately Demented") 
-            st.markdown(f" Prediction Probability: {rounded_probability}")
+            st.markdown(f" Prediction Probability: {probability}")
                      
         elif predict_val == 2:
-            probability = pred[0][predict_val]
-            rounded_probability = "{:.3f}".format(probability)
+            probability = pred[0][predict_val][0]            
             st.markdown(" Stage: Not Demented") 
-            st.markdown(f" Prediction Probability: {rounded_probability}")
+            st.markdown(f" Prediction Probability: {probability}")
                     
         elif predict_val == 3:
-            probability = pred[0][predict_val]
-            rounded_probability = "{:.3f}".format(probability)
+            probability = pred[0][predict_val][0]            
             st.markdown(" Stage: Very Mildly Demented") 
-            st.markdown(f" Prediction Probability: {rounded_probability}")
+            st.markdown(f" Prediction Probability: {probability}")
                     
         else:
            st.warning("Error!")