Update app.py

app.py

@@ -1,5 +1,6 @@
 # -*- coding: utf-8 -*-
 # Copyright 2024-2025 The Alibaba Wan Team Authors. All rights reserved
+import spaces
 import logging,os
 os.makedirs("/root/weights",exist_ok=True)
 cmd="huggingface-cli download IndexTeam/AnisoraV3 --include=\"14B/*\" --local-dir=/root/weights --token %s"%os.environ['token']
@@ -256,6 +257,7 @@ def generate(args):
         t5_cpu=args.t5_cpu,
     )
 
+    @spaces.GPU
     def generate_i2v(prompt,img,seed,nf,speed):
         logging.info("Generating video ...")
         save_file="output/%s-%s-%s-%s.mp4"%(seed,nf,speed,int(ttime()))