Update app.py

app.py

@@ -190,7 +190,7 @@ def main():
         
         # Display original image
         st.subheader("Original Image")
-        st.image(image, use_column_width=True)
+        st.image(image, use_container_width=True)
         
         # Preprocess the image
         img_array = explainer.preprocess_image(image)
@@ -215,12 +215,12 @@ def main():
             
             with col1:
                 st.subheader("Grad-CAM Heatmap")
-                st.image(heatmap, use_column_width=True, 
+                st.image(heatmap, use_container_width=True, 
                          caption="Areas of model's focus (red = high importance)")
             
             with col2:
                 st.subheader("Heatmap Overlay")
-                st.image(superimposed_img, use_column_width=True, 
+                st.image(superimposed_img, use_container_width=True, 
                          caption="Heatmap superimposed on original image")
             
             # Explanation of the visualization