change to cpu

.gitignore

@@ -4,4 +4,5 @@ temp_output/
 processed_images.zip
 test_input/
 test_output/
-bria_output/
+bria_output/
+processed_images/

app.py

@@ -22,7 +22,7 @@ def remove_background_rembg(input_path):
 
 def remove_background_bria(input_path):
     print(f"Removing background using bria for image: {input_path}")
-    pipe = pipeline("image-segmentation", model="briaai/RMBG-1.4", trust_remote_code=True, device=0)
+    pipe = pipeline("image-segmentation", model="briaai/RMBG-1.4", trust_remote_code=True, device=-1)
     pillow_image = pipe(input_path)
     return pillow_image