Update app.py

app.py

@@ -39,6 +39,8 @@ from tqdm import tqdm
 from models.multimodal_encoder.builder import build_image_tower, build_video_tower
 from models.multimodal_projector.builder import build_vision_projector
 
+import spaces
+
 
 title_markdown = ("""<div class="embed_hidden" style="text-align: center;">
     <h1>MotionLLM: Understanding Human Behaviors from Human Motions and Videos</h1>
@@ -295,7 +297,7 @@ def get_processor(X, config, device, pretrained_checkpoint_tower, model_path = '
 
     return mm_backbone_mlp_model, processor
 
-
+@spaces.GPU
 def motionllm(
     args, 
     input_video_path: str,
@@ -361,7 +363,7 @@ def save_video_to_local(video_path):
     shutil.copyfile(video_path, filename)
     return filename
 
-
+@spaces.GPU
 def generate(image1, video, textbox_in, first_run, state, images_tensor):
     flag = 1