Add application file

app.py

@@ -12,6 +12,7 @@ from wan_pipeline import WanPipeline
 from diffusers.schedulers.scheduling_unipc_multistep import UniPCMultistepScheduler
 from PIL import Image
 from diffusers.utils import export_to_video
+import spaces
 
 def set_seed(seed):
     random.seed(seed)
@@ -52,6 +53,7 @@ def load_model(model_name):
     return current_model.to('cuda')
 
 
+@spaces.GPU(duration=120)
 def generate_content(prompt, model_name, guidance_scale=7.5, num_inference_steps=50, use_cfg_zero_star=True, use_zero_init=True, zero_steps=0, seed=None, compare_mode=False):
     model = load_model(model_name)
     if seed is None:

requirements.txt

@@ -11,4 +11,4 @@ moviepy
 sentencepiece
 Pillow==9.5.0
 ftfy
-
+spaces