freddyaboulton HF staff commited on
Commit
080458f
·
verified ·
1 Parent(s): c8120da

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. inference.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ from pydantic import BaseModel, Field
11
 
12
  try:
13
  from demo.object_detection.inference import YOLOv10
14
- except ImportError:
15
  from .inference import YOLOv10
16
 
17
 
 
11
 
12
  try:
13
  from demo.object_detection.inference import YOLOv10
14
+ except (ImportError, ModuleNotFoundError):
15
  from .inference import YOLOv10
16
 
17
 
inference.py CHANGED
@@ -6,7 +6,7 @@ import onnxruntime
6
 
7
  try:
8
  from demo.object_detection.utils import draw_detections
9
- except ImportError:
10
  from .utils import draw_detections
11
 
12
 
 
6
 
7
  try:
8
  from demo.object_detection.utils import draw_detections
9
+ except (ImportError, ModuleNotFoundError):
10
  from .utils import draw_detections
11
 
12