Update app.py

app.py

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