Update app.py

app.py

@@ -210,9 +210,7 @@ def infer(
         for img in gallery_images:
             image = img[0]  # Extract PIL image from gallery format
             
-            # Apply background removal only for reference mode if checkbox is checked
-            if mode == "Reference" and remove_bg:  # Updated to use new mode name
-                image = remove_background_from_image(image)
+            image = remove_background_from_image(image)
             
             # Always remove alpha channels to ensure RGB format
             image = remove_alpha_channel(image)