onescotch commited on
Commit
85d8725
1 Parent(s): 0515b76

extend gpu usage time

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ print(torch.cuda.get_device_name(index))
27
  # from main.inference import Inferer
28
  # inferer = Inferer(DEFAULT_MODEL, num_gpus, OUT_FOLDER)
29
 
30
- @spaces.GPU(enable_queue=True)
31
  def infer(video_input, in_threshold=0.5, num_people="Single person", render_mesh=False):
32
  from main.inference import Inferer
33
  inferer = Inferer(DEFAULT_MODEL, num_gpus, OUT_FOLDER)
 
27
  # from main.inference import Inferer
28
  # inferer = Inferer(DEFAULT_MODEL, num_gpus, OUT_FOLDER)
29
 
30
+ @spaces.GPU(enable_queue=True, duration=300)
31
  def infer(video_input, in_threshold=0.5, num_people="Single person", render_mesh=False):
32
  from main.inference import Inferer
33
  inferer = Inferer(DEFAULT_MODEL, num_gpus, OUT_FOLDER)