Update app.py

app.py

@@ -494,10 +494,10 @@ if uploaded_file is not None:
     with col3:
         if st.button('Generate Grad-CAM'):
             st.write("Loading model...")
-            model = load_model()
+            model_gradcam = keras.models.load_model('./gradcam_model.h5')
             # Compute and show Grad-CAM
             st.write("Generating Grad-CAM visualizations")
             try:
-                compute_gradcam(model, uploaded_file)
+                compute_gradcam(model_gradcam, uploaded_file)
             except Exception as e:
                 st.error(f"Error generating Grad-CAM: {e}")