Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
2e513bf
1
Parent(s):
d4742c0
Enable queue for spaces
Browse files
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]
|