tuansunday05 commited on
Commit
2e37c6a
·
verified ·
1 Parent(s): 9432225

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
@@ -15,7 +15,7 @@ from scipy.interpolate import make_interp_spline
15
  import pandas as pd
16
  should_continue = True
17
 
18
- # @spaces.GPU(duration=240)
19
  def yolov9_inference(model_id, img_path=None, vid_path=None, tracking_algorithm = None):
20
  global should_continue
21
  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
 
15
  import pandas as pd
16
  should_continue = True
17
 
18
+ @spaces.GPU(duration=240)
19
  def yolov9_inference(model_id, img_path=None, vid_path=None, tracking_algorithm = None):
20
  global should_continue
21
  img_extensions = ['.jpg', '.jpeg', '.png', '.gif'] # Add more image extensions if needed