tuansunday05 commited on
Commit
68e3bf5
·
verified ·
1 Parent(s): 21813a3

Integrate spaces for running on ZeroGPU

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -1,3 +1,4 @@
 
1
  import gradio as gr
2
  from detect_strongsort import run
3
  import os
@@ -33,6 +34,7 @@ def yolov9_inference(model_id, image_size, conf_threshold, iou_threshold, img_pa
33
 
34
  return output_image, output_video, output_path
35
 
 
36
  def inference(model_id, image_size, conf_threshold, iou_threshold, img_path=None, vid_path=None):
37
  global should_continue
38
  should_continue = True
 
1
+ import spaces
2
  import gradio as gr
3
  from detect_strongsort import run
4
  import os
 
34
 
35
  return output_image, output_video, output_path
36
 
37
+ @spaces.GPU(duration=120)
38
  def inference(model_id, image_size, conf_threshold, iou_threshold, img_path=None, vid_path=None):
39
  global should_continue
40
  should_continue = True