Shengxiao0709 commited on
Commit
a3ef2be
·
verified ·
1 Parent(s): d6ea8cb

Update inference_cout.py

Browse files
Files changed (1) hide show
  1. inference_cout.py +1 -1
inference_cout.py CHANGED
@@ -8,7 +8,7 @@ import matplotlib.pyplot as plt
8
  import tempfile
9
  import os
10
  from huggingface_hub import hf_hub_download
11
- from counting_model import CountingModule
12
 
13
  MODEL = None
14
  DEVICE = torch.device("cpu")
 
8
  import tempfile
9
  import os
10
  from huggingface_hub import hf_hub_download
11
+ from counting import CountingModule
12
 
13
  MODEL = None
14
  DEVICE = torch.device("cpu")