Update app.py

app.py

@@ -125,7 +125,7 @@ def main():
             try:
                 # Display image preview
                 image = Image.open(original_image)
-                st.image(image, caption="Original Image", use_column_width=True)
+                st.image(image, caption="Original Image", use_container_width=True)
                 st.caption(f"Original size: {original_image.size/1024/1024:.2f}MB")
             except Exception as e:
                 st.error(f"Error loading image: {str(e)}")