updates

app.py

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