Update app.py

app.py

@@ -159,7 +159,7 @@ def medical_page():
             resized_img = resized_img[np.newaxis, ...]
             class_names = ['ModerateDemented', 'NonDemented', 'VeryMildDemented', 'MildDemented']
             
-            pred = class_names[np.argmax(st.session_state.alzheimer_model.predict(resized_img))]
+            pred = class_names[np.argmax(alzheimer_model.predict(resized_img))]
             st.text(pred)
 
     def brain_tumor():
@@ -181,7 +181,7 @@ def medical_page():
             resized_img = resized_img[np.newaxis, ...]
             class_names = ['Pituitary', 'No-Tumor', 'Meningioma', 'Glioma']
             
-            pred = class_names[np.argmax(st.session_state.brain_tumor_model.predict(resized_img))]
+            pred = class_names[np.argmax(brain_tumor_model.predict(resized_img))]
             st.text(pred)
 
     def abscesses():