add GPU

app.py

@@ -58,6 +58,7 @@ import gradio as gr
 from pathlib import Path
 from urllib.parse import urlparse
 from torch.hub import download_url_to_file, get_dir
+from spaces import GPU
 
 
 def load_file_from_url(url, model_dir=None, progress=True, file_name=None):
@@ -199,6 +200,7 @@ def generation_step(runner, text_embeds_dict, cond_latents):
 
     return samples
 
+@GPU
 def generation_loop(video_path='./test_videos', output_dir='./results', seed=666, batch_size=1, cfg_scale=1.0, cfg_rescale=0.0, sample_steps=1, res_h=1280, res_w=720, sp_size=1):
     runner = configure_runner(1)
     output_dir = 'output/out.mp4'