Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
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],
|