Spaces:
Sleeping
Sleeping
Update inference_cout.py
Browse files- 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
|
| 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")
|