Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -12,7 +12,7 @@ import os
|
|
12 |
import tempfile
|
13 |
import uuid
|
14 |
|
15 |
-
torch.set_float32_matmul_precision("
|
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
|