bulatko commited on
Commit
127405e
·
1 Parent(s): c3a1d9d

Reduce GPU duration to 120s to be more efficient with Zero GPU quota

Browse files
Files changed (1) hide show
  1. mvp.py +1 -1
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=300) # 5 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.
 
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.