bug fix

app.py

@@ -39,13 +39,14 @@ st.markdown("[Dataset Source](https://www.kaggle.com/datasets/tourist55/alzheime
 model = tf.keras.models.load_model('./model/model_1.h5')
 model.load_weights('./model/best_model_custom_1.h5')
 
-uploaded_file = st.file_uploader("Upload a brain MRI image here", type=["jpg", "png", "jpeg"], label_visibility="hidden")
-predict_button = st.button("ㅤㅤPredictㅤㅤ")
+uploaded_file = st.file_uploader("Upload a brain MRI image here", type=["jpg", "png", "jpeg"])
 
 if uploaded_file is not None:
     file_bytes = BytesIO(uploaded_file.read())
     st.image(file_bytes,use_column_width=True,clamp = True) 
     
+predict_button = st.button("ㅤㅤPredictㅤㅤ")
+if predict_button:
     img = cv2.imdecode(np.frombuffer(file_bytes.read(), np.uint8), 0)
     #img=np.array(file_bytes)   
   
@@ -61,19 +62,19 @@ if uploaded_file is not None:
     predict_val = np.argmax(pred, axis=1)
    
     if predict_val == 0:
-        probability = pred[predict_val]
+        probability = pred[0][predict_val]
         st.write(f"Mildly Demented with prediction probability of {probability}") 
                  
     elif predict_val == 1:
-        probability = pred[predict_val]
+        probability = pred[0][predict_val]
         st.write(f"Moderately Demented with prediction probability of {probability}") 
                  
     elif predict_val == 2:
-        probability = pred[predict_val]
+        probability = pred[0][predict_val]
         st.write(f"Not Demented with prediction probability of {probability}") 
                 
     elif predict_val == 3:
-        probability = pred[predict_val]
+        probability = pred[0][predict_val]
         st.write(f"Very Mildly Demented with prediction probability of {probability}") 
                 
     else: