ntsc207 commited on
Commit
8d42b57
·
verified ·
1 Parent(s): 70a5a36

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,4 +1,4 @@
1
- #import spaces
2
  import gradio as gr
3
  from detect_deepsort import run_deepsort
4
  from detect_strongsort import run_strongsort
@@ -14,7 +14,7 @@ import threading
14
 
15
  should_continue = True
16
 
17
- #@spaces.GPU(duration=120)
18
  def yolov9_inference(model_id, img_path=None, vid_path=None, tracking_algorithm = None):
19
  global should_continue
20
  img_extensions = ['.jpg', '.jpeg', '.png', '.gif'] # Add more image extensions if needed
 
1
+ import spaces
2
  import gradio as gr
3
  from detect_deepsort import run_deepsort
4
  from detect_strongsort import run_strongsort
 
14
 
15
  should_continue = True
16
 
17
+ @spaces.GPU(duration=120)
18
  def yolov9_inference(model_id, img_path=None, vid_path=None, tracking_algorithm = None):
19
  global should_continue
20
  img_extensions = ['.jpg', '.jpeg', '.png', '.gif'] # Add more image extensions if needed