Update app.py

app.py

@@ -770,8 +770,8 @@ def main():
             with col1:
                 # Display original image and overlay side by side with controlled size
                 st.image(st.session_state.current_image, caption="Original Image", width=300)
-                if hasattr(st.session_state, 'current_overlay'):
-                    st.image(st.session_state.current_overlay, caption="GradCAM Overlay", width=300)
+                # if hasattr(st.session_state, 'current_overlay'):
+                #     st.image(st.session_state.current_overlay, caption="GradCAM Overlay", width=300)
             
             with col2:
                 # Image description
@@ -781,9 +781,9 @@ def main():
                     st.markdown("---")
                 
                 # GradCAM analysis
-                if hasattr(st.session_state, 'gradcam_caption'):
-                    st.markdown("### GradCAM Analysis")
-                    st.markdown(st.session_state.gradcam_caption)
+                # if hasattr(st.session_state, 'gradcam_caption'):
+                #     st.markdown("### GradCAM Analysis")
+                #     st.markdown(st.session_state.gradcam_caption)
     
     # LLM Analysis section - AFTER Image Analysis Summary
     with st.expander("Stage 3: Detailed Analysis with Vision LLM", expanded=False):