Spaces:

mrdilaw
/

BACKROUND

Runtime error

mrdilaw commited on May 27

Commit

5cc7c4c

verified ·

1 Parent(s): 6634469

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ 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
 )
+birefnet.to("cpu")
 # تجهيز الصورة قبل الإدخال
 transform_image = transforms.Compose([