Update app.py

app.py

@@ -121,7 +121,7 @@ def main():
                 image = image.convert('RGB')
             
             col1.subheader("Original Image")
-            col1.image(image, use_column_width=True)
+            col1.image(image, use_container_width=True)  # Updated parameter
             
             # Process image and get predictions
             with st.spinner('Processing image...'):
@@ -133,7 +133,7 @@ def main():
                 
                 # Display annotated image
                 col2.subheader("Detected Conditions")
-                col2.image(image_with_boxes, use_column_width=True)
+                col2.image(image_with_boxes, use_container_width=True)  # Updated parameter
                 
                 # Display predictions in a nice format
                 st.subheader("Detailed Results")