Reduce GPU duration to 120s to be more efficient with Zero GPU quota
Browse files
mvp.py
CHANGED
|
@@ -1126,7 +1126,7 @@ def visualize_detections(target_dir, conf_thres, frame_filter="All", mask_black_
|
|
| 1126 |
|
| 1127 |
return out_path, legend_md
|
| 1128 |
|
| 1129 |
-
@spaces.GPU(duration=
|
| 1130 |
def detect_objects(text_labels, target_dir, conf_thres, *viz_args):
|
| 1131 |
"""
|
| 1132 |
Detect objects from text labels and return the detected objects.
|
|
|
|
| 1126 |
|
| 1127 |
return out_path, legend_md
|
| 1128 |
|
| 1129 |
+
@spaces.GPU(duration=120) # 2 minutes for full pipeline on Zero GPU
|
| 1130 |
def detect_objects(text_labels, target_dir, conf_thres, *viz_args):
|
| 1131 |
"""
|
| 1132 |
Detect objects from text labels and return the detected objects.
|