Spaces:
Running
on
Zero
Running
on
Zero
kxhit
commited on
Commit
·
d161cfd
1
Parent(s):
11fadb1
carvekit cpu
Browse files
app.py
CHANGED
@@ -136,7 +136,7 @@ def create_carvekit_interface():
|
|
136 |
interface = HiInterface(object_type="object", # Can be "object" or "hairs-like".
|
137 |
batch_size_seg=6,
|
138 |
batch_size_matting=1,
|
139 |
-
device=
|
140 |
seg_mask_size=640, # Use 640 for Tracer B7 and 320 for U2Net
|
141 |
matting_mask_size=2048,
|
142 |
trimap_prob_threshold=231,
|
|
|
136 |
interface = HiInterface(object_type="object", # Can be "object" or "hairs-like".
|
137 |
batch_size_seg=6,
|
138 |
batch_size_matting=1,
|
139 |
+
device="cpu",
|
140 |
seg_mask_size=640, # Use 640 for Tracer B7 and 320 for U2Net
|
141 |
matting_mask_size=2048,
|
142 |
trimap_prob_threshold=231,
|