dramp77 commited on
Commit
16e01a4
·
verified ·
1 Parent(s): 501279e

change device to cpu

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=0)
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