Update app.py

app.py

@@ -1042,13 +1042,13 @@ def main():
             
             with col_images:
                 st.write("#### Reference Images")
-                st.image(st.session_state.current_image, caption="Original", use_column_width=True)
+                st.image(st.session_state.current_image, caption="Original", use_container_width=True)
                 
                 if hasattr(st.session_state, 'current_overlay'):
-                    st.image(st.session_state.current_overlay, caption="GradCAM", use_column_width=True)
+                    st.image(st.session_state.current_overlay, caption="GradCAM", use_container_width=True)
                 
                 if hasattr(st.session_state, 'comparison_image'):
-                    st.image(st.session_state.comparison_image, caption="Comparison", use_column_width=True)
+                    st.image(st.session_state.comparison_image, caption="Comparison", use_container_width=True)
                     
                 if hasattr(st.session_state, 'current_pred_label'):
                     st.info(f"**Classification:** {st.session_state.current_pred_label} (Confidence: {st.session_state.current_confidence:.2%})")