multimodalart HF staff commited on
Commit
2e513bf
1 Parent(s): d4742c0

Enable queue for spaces

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -479,7 +479,7 @@ def start_training_og(
479
  train_main(args)
480
  return "ok!"
481
 
482
- @spaces.GPU()
483
  def run_captioning(*inputs):
484
  model.to("cuda")
485
  images = inputs[0]
 
479
  train_main(args)
480
  return "ok!"
481
 
482
+ @spaces.GPU(enable_queue=True)
483
  def run_captioning(*inputs):
484
  model.to("cuda")
485
  images = inputs[0]