Update app.py

app.py

@@ -72,6 +72,9 @@ if url:
         if response.status_code == 200:
             image = Image.open(BytesIO(response.content))
             st.sidebar.image(image.resize((200, 200)), caption="Uploaded Image", use_column_width=False)
+            # Ensure the image is in RGBA mode (or RGB)
+            if image.mode != 'RGB':
+                image = image.convert('RGB')
             rmimg = remove(image)
             # Convert image to numpy array for YOLO model
             image_np = np.array(image)