Spaces:

chuuhtetnaing
/

background-remover

Running

chuuhtetnaing commited on Nov 2, 2024

Commit

3618356

1 Parent(s): f0a6ca3

changed to cpu

Files changed (1) hide show

app.py CHANGED Viewed

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

 birefnet = AutoModelForImageSegmentation.from_pretrained(
     "ZhengPeng7/BiRefNet", trust_remote_code=True
 )
+birefnet.to("cpu")
 transform_image = transforms.Compose(
     [
         transforms.Resize((1024, 1024)),