Update app.py

app.py

@@ -56,7 +56,7 @@ uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "png", "jpeg
 if uploaded_file:
     # Read and display the image
     image = Image.open(uploaded_file)
-    st.image(image, caption="Uploaded Image", use_column_width=True)
+    st.image(image, caption="Uploaded Image", use_container_width=True)
 
     # Perform prediction
     with st.spinner("Processing..."):
@@ -67,6 +67,6 @@ if uploaded_file:
                 iou=overlap_threshold,  # Set overlap (IoU) threshold
             )
             st.write("Detection Results:")
-            st.image(results[0].plot(), caption="Detections", use_column_width=True)
+            st.image(results[0].plot(), caption="Detections", use_container_width=True)
         except Exception as e:
             st.error(f"Error during prediction: {e}")