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]
|