Spaces:
Runtime error
Runtime error
Update inference.py
Browse files- inference.py +1 -1
inference.py
CHANGED
|
@@ -16,7 +16,7 @@ from demo_utils.draw import center, draw
|
|
| 16 |
|
| 17 |
def inference(
|
| 18 |
input_video: str,
|
| 19 |
-
model: str = "YOLOv7",
|
| 20 |
features: str = [0, 1],
|
| 21 |
track_points: str = "Bounding box",
|
| 22 |
model_threshold: float = 0.25,
|
|
|
|
| 16 |
|
| 17 |
def inference(
|
| 18 |
input_video: str,
|
| 19 |
+
model: str = "YOLOv7 Tiny",
|
| 20 |
features: str = [0, 1],
|
| 21 |
track_points: str = "Bounding box",
|
| 22 |
model_threshold: float = 0.25,
|