Hugging Face's logo

Spaces:
phoebe777777
/
FinalVision
Sleeping

App Files Community
Shengxiao0709 commited on
Commit
04954a4
·
verified ·
1 Parent(s): f409b59

Update inference_seg.py

Browse files
Files changed (1) hide show
  1. inference_seg.py +1 -1
inference_seg.py CHANGED
@@ -8,7 +8,7 @@ DEVICE = torch.device("cpu")
8
 
9
  def load_model(use_box=False):
10
  global MODEL, DEVICE
11
- MODEL = CountingModule(use_box=use_box)
12
 
13
  ckpt_path = hf_hub_download(
14
  repo_id="Shengxiao0709/cellsegmodel",
 
8
 
9
  def load_model(use_box=False):
10
  global MODEL, DEVICE
11
+ MODEL = SegmentationModule(use_box=use_box)
12
 
13
  ckpt_path = hf_hub_download(
14
  repo_id="Shengxiao0709/cellsegmodel",