Hugging Face's logo

Spaces:
Ash2505
/
EEE515-HW3
Sleeping

App Files Community
Ash2505 commited on
Commit
f76d996
·
verified ·
1 Parent(s): 613145b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
12
 
13
  birefnet = AutoModelForImageSegmentation.from_pretrained('ZhengPeng7/BiRefNet', trust_remote_code=True)
14
  torch.set_float32_matmul_precision(['high', 'highest'][0])
15
- birefnet.to('cuda')
16
  birefnet.eval()
17
  birefnet.half()
18
 
 
12
 
13
  birefnet = AutoModelForImageSegmentation.from_pretrained('ZhengPeng7/BiRefNet', trust_remote_code=True)
14
  torch.set_float32_matmul_precision(['high', 'highest'][0])
15
+ birefnet.to(device)
16
  birefnet.eval()
17
  birefnet.half()
18