Update app.py

app.py

@@ -12,12 +12,6 @@ from classification import pipes
 from rembg import remove
 from PIL import Image
 
-# input_path = 'input.png'
-# output_path = 'output.png'
-
-# input = Image.open(input_path)
-# output = remove(input)
-# output.save(output_path)
 time_taken={}
 # Categories dictionary
 categories_dict = {
@@ -73,9 +67,9 @@ if url:
             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)
+            if rmimg.mode != 'RGB':
+                rmimg = rmimg.convert('RGB')
             # Convert image to numpy array for YOLO model
             image_np = np.array(image)
             outputs={}