feat: introduce `rembg` library for background removal functionality

app.py

@@ -96,6 +96,7 @@ def extract_frames_and_masks(video_path, garment_name):
         
         # Remove background (get mask)
         # rembg returns RGBA, alpha channel is the mask
+        from rembg import remove
         output = remove(pil_im)
         mask = np.array(output)[:, :, 3] # Extract alpha
         

requirements.txt

@@ -1,6 +1,6 @@
 gradio
 opencv-python-headless
-onnxruntime-gpu
+onnxruntime
 wheel
 ninja
 setuptools<70