Update app.py

app.py

@@ -297,6 +297,7 @@ def get_processor(X, config, device, pretrained_checkpoint_tower, model_path = '
 
     return mm_backbone_mlp_model, processor
 
+
 @spaces.GPU
 def motionllm(
     args, 
@@ -346,7 +347,8 @@ def motionllm(
     print("================================")
     print(output)
     print("================================")
-    
+
+    prompt = (prompt[0].cpu(), prompt[1])
     return output, input_prompt, prompt
 
 
@@ -363,7 +365,7 @@ def save_video_to_local(video_path):
     shutil.copyfile(video_path, filename)
     return filename
 
-# @spaces.GPU
+@spaces.GPU
 def generate(image1, video, textbox_in, first_run, state, images_tensor):
     flag = 1