Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Commit
•
0b00f1f
1
Parent(s):
bfef560
Update app.py
Browse files
app.py
CHANGED
@@ -129,7 +129,7 @@ def run_captioning(images, concept_sentence, *captions):
|
|
129 |
del processor
|
130 |
|
131 |
if is_spaces:
|
132 |
-
|
133 |
|
134 |
def start_training(
|
135 |
profile: Union[gr.OAuthProfile, None],
|
|
|
129 |
del processor
|
130 |
|
131 |
if is_spaces:
|
132 |
+
run_captioning = spaces.GPU()(run_captioning)
|
133 |
|
134 |
def start_training(
|
135 |
profile: Union[gr.OAuthProfile, None],
|