ZhenweiWang commited on
Commit
4c5a78b
·
verified ·
1 Parent(s): 0ca05b5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -75,7 +75,7 @@ class TeeOutput:
75
  # -------------------------------------------------------------------------
76
  # Model inference
77
  # -------------------------------------------------------------------------
78
- @spaces.GPU()
79
  def run_model(
80
  target_dir,
81
  confidence_percentile: float = 10,
@@ -466,7 +466,7 @@ def prepare_visualization_data(
466
 
467
  return visualization_dict
468
 
469
- @spaces.GPU()
470
  def gradio_demo(
471
  target_dir,
472
  frame_selector="All",
 
75
  # -------------------------------------------------------------------------
76
  # Model inference
77
  # -------------------------------------------------------------------------
78
+ @spaces.GPU(duration=120)
79
  def run_model(
80
  target_dir,
81
  confidence_percentile: float = 10,
 
466
 
467
  return visualization_dict
468
 
469
+ @spaces.GPU(duration=120)
470
  def gradio_demo(
471
  target_dir,
472
  frame_selector="All",