Spaces:

joermd
/

removeback

Sleeping

joermd commited on Nov 18, 2024

Commit

bcb4492

verified ·

1 Parent(s): 87e1933

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,6 @@ torch.set_float32_matmul_precision("high")
 birefnet = AutoModelForImageSegmentation.from_pretrained(
     "ZhengPeng7/BiRefNet", trust_remote_code=True
 )
-birefnet.to("cuda")
 transform_image = transforms.Compose(
     [

 birefnet = AutoModelForImageSegmentation.from_pretrained(
     "ZhengPeng7/BiRefNet", trust_remote_code=True
 )
 transform_image = transforms.Compose(
     [