update

app.py

@@ -1,3 +1,4 @@
+import spaces
 import warnings
 warnings.filterwarnings("ignore")
 
@@ -50,6 +51,7 @@ def create_pipeline(device=torch.device("cuda"), dtype=torch.float32):
 pipeline, category_text_encoding_mapping = create_pipeline(device, dtype)
 
 
+@spaces.GPU(duration=120)
 def generate_video(image, audio, text, audio_guidance_scale, denoising_step):
 	
 	category_text_encoding = category_text_encoding_mapping[text].view(1, 77, 768)