Update app.py

app.py

@@ -30,6 +30,7 @@ task_type = None
 def get_transformer_model_id(task_type:str) -> str:
     return "Skywork/SkyReels-V1-Hunyuan-I2V" if task_type == "i2v" else "Skywork/SkyReels-V1-Hunyuan-T2V"
 
+@spaces.GPU()
 def init_predictor(task_type:str, gpu_num:int=1):
     global predictor
     predictor = SkyReelsVideoInfer(
@@ -44,7 +45,8 @@ def init_predictor(task_type:str, gpu_num:int=1):
             compiler_transformer=False,
         )
     )
-
+    
+@spaces.GPU(duration=90)
 def generate_video(prompt, seed, image=None):
     global task_type
     print(f"image:{type(image)}")