Upload app.py

app.py

@@ -242,7 +242,7 @@ def main():
             
             with col1:
                 st.markdown("#### Original Image")
-                st.image(image_rgb, use_container_width=True)
+                st.image(image_rgb, use_column_width=True)
             
             # Run detection
             with st.spinner("🔍 Detecting garbage..."):
@@ -255,7 +255,7 @@ def main():
             with col2:
                 st.markdown("#### Detection Results")
                 annotated_rgb = cv2.cvtColor(result["image"], cv2.COLOR_BGR2RGB)
-                st.image(annotated_rgb, use_container_width=True)
+                st.image(annotated_rgb, use_column_width=True)
             
             # Display statistics
             st.markdown("---")