Spaces:
Runtime error
Runtime error
Commit
·
0cb2d2a
1
Parent(s):
547bded
fix: cache examples
Browse files- app.py +1 -1
- inference.py +4 -13
app.py
CHANGED
|
@@ -44,7 +44,7 @@ iface = gr.Interface(
|
|
| 44 |
inputs=intput_components,
|
| 45 |
outputs=output_components,
|
| 46 |
examples=example_list,
|
| 47 |
-
cache_examples=
|
| 48 |
)
|
| 49 |
|
| 50 |
iface.launch()
|
|
|
|
| 44 |
inputs=intput_components,
|
| 45 |
outputs=output_components,
|
| 46 |
examples=example_list,
|
| 47 |
+
cache_examples=True,
|
| 48 |
)
|
| 49 |
|
| 50 |
iface.launch()
|
inference.py
CHANGED
|
@@ -19,22 +19,13 @@ from demo_utils.files import get_files
|
|
| 19 |
|
| 20 |
def inference(
|
| 21 |
input_video: str,
|
| 22 |
-
|
| 23 |
-
|
| 24 |
-
|
| 25 |
-
|
| 26 |
-
model_threshold: str,
|
| 27 |
):
|
| 28 |
# temp_dir = tempfile.TemporaryDirectory()
|
| 29 |
# output_path = temp_dir.name
|
| 30 |
-
|
| 31 |
-
if cb_cache_output:
|
| 32 |
-
for cache_video in get_files("examples"):
|
| 33 |
-
if input_video.find(cache_video) == -1:
|
| 34 |
-
continue
|
| 35 |
-
|
| 36 |
-
cache_video_name = f"outputs/{cache_video}_{model}_out.mp4"
|
| 37 |
-
return cache_video_name
|
| 38 |
|
| 39 |
coord_transformations = None
|
| 40 |
paths_drawer = None
|
|
|
|
| 19 |
|
| 20 |
def inference(
|
| 21 |
input_video: str,
|
| 22 |
+
model: str = "YOLOv7",
|
| 23 |
+
features: str = [0, 1],
|
| 24 |
+
track_points: str = "Bounding box",
|
| 25 |
+
model_threshold: float = 0.25,
|
|
|
|
| 26 |
):
|
| 27 |
# temp_dir = tempfile.TemporaryDirectory()
|
| 28 |
# output_path = temp_dir.name
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 29 |
|
| 30 |
coord_transformations = None
|
| 31 |
paths_drawer = None
|