Update app.py

app.py

@@ -71,11 +71,6 @@ def load_captioning(uploaded_images, concept_sentence):
     updates.append(gr.update(placeholder=f"A {concept_sentence} in a mall"))
     return updates
 
-
-if is_spaces:
-    load_captioning = spaces.GPU()(load_captioning)
-
-
 def create_dataset(*inputs):
     print("Creating dataset")
     images = inputs[0]
@@ -133,6 +128,8 @@ def run_captioning(images, concept_sentence, *captions):
     del model
     del processor
 
+if is_spaces:
+    load_captioning = spaces.GPU()(run_captioning)
 
 def start_training(
     profile: Union[gr.OAuthProfile, None],