Update app.py

app.py

@@ -20,6 +20,7 @@ from pathlib import Path
 from datetime import datetime, timezone
 
 import torch
+import spaces
 import numpy as np
 import cv2
 from PIL import Image
@@ -231,7 +232,7 @@ def generate_video_infer(video_path, prompt_text, max_new_tokens=1024, temperatu
 # ---------------------------
 # Unified generation function for ChatInterface
 # ---------------------------
-
+@spaces.GPU
 def generate(input_dict: dict, chat_history: list[dict],
              max_new_tokens: int, temperature: float,
              top_p: float, top_k: int, repetition_penalty: float):