change device to cpu
Browse files
app.py
CHANGED
@@ -22,7 +22,7 @@ def remove_background_rembg(input_path):
|
|
22 |
|
23 |
def remove_background_bria(input_path):
|
24 |
print(f"Removing background using bria for image: {input_path}")
|
25 |
-
pipe = pipeline("image-segmentation", model="briaai/RMBG-1.4", trust_remote_code=True, device=
|
26 |
pillow_image = pipe(input_path)
|
27 |
return pillow_image
|
28 |
|
|
|
22 |
|
23 |
def remove_background_bria(input_path):
|
24 |
print(f"Removing background using bria for image: {input_path}")
|
25 |
+
pipe = pipeline("image-segmentation", model="briaai/RMBG-1.4", trust_remote_code=True, device='cpu')
|
26 |
pillow_image = pipe(input_path)
|
27 |
return pillow_image
|
28 |
|