Commit
•
4dcc0c1
1
Parent(s):
db9c00f
Update app.py
Browse files
app.py
CHANGED
|
@@ -430,7 +430,9 @@ def start_training_og(
|
|
| 430 |
#subprocess.run(commands)
|
| 431 |
return "ok!"
|
| 432 |
|
|
|
|
| 433 |
def run_captioning(*inputs):
|
|
|
|
| 434 |
print(inputs)
|
| 435 |
images = inputs[0]
|
| 436 |
training_option = inputs[-1]
|
|
|
|
| 430 |
#subprocess.run(commands)
|
| 431 |
return "ok!"
|
| 432 |
|
| 433 |
+
@spaces.GPU()
|
| 434 |
def run_captioning(*inputs):
|
| 435 |
+
model.to("cuda")
|
| 436 |
print(inputs)
|
| 437 |
images = inputs[0]
|
| 438 |
training_option = inputs[-1]
|