Update app.py

app.py

@@ -135,11 +135,7 @@ if len(file_upload) == 1:
                 'MCEBNfold3.pth', map_location='cpu'))
             model.eval()
             prediction = model(test_images.unsqueeze(1))
-            st.write("deeev")#re
-            st.write(prediction)#re
             pred = prediction.argmax(dim=1).item()
-            st.write("gfhgfhdeev")#rem
-            st.write(pred)#rem
             class_names = ["SMCI", "AD", "CN"]
             predicted_label = class_names[pred]