KingNish commited on
Commit
0af2d38
1 Parent(s): a2b9299

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ import os
12
  import tempfile
13
  import uuid
14
 
15
- torch.set_float32_matmul_precision("highest")
16
 
17
  birefnet = AutoModelForImageSegmentation.from_pretrained(
18
  "ZhengPeng7/BiRefNet", trust_remote_code=True
 
12
  import tempfile
13
  import uuid
14
 
15
+ torch.set_float32_matmul_precision("medium")
16
 
17
  birefnet = AutoModelForImageSegmentation.from_pretrained(
18
  "ZhengPeng7/BiRefNet", trust_remote_code=True