kxhit commited on
Commit
d161cfd
·
1 Parent(s): 11fadb1

carvekit cpu

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=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,