Update app.py

app.py

@@ -15,7 +15,8 @@ def get_transformer_model_id(task_type: str) -> str:
         return "Skywork/skyreels-v1-Hunyuan-i2v"
     else:
         return "Skywork/skyreels-v1-Hunyuan-t2v"
-
+        
+@spaces.GPU(duration=120)
 def init_predictor(task_type: str):
     # ALL IMPORTS NOW INSIDE THIS FUNCTION
     import torch
@@ -43,7 +44,7 @@ def init_predictor(task_type: str):
     except Exception as e:
         return f"Error loading model: {e}", None
 
-
+@spaces.GPU(duration=80)
 def generate_video(prompt, seed, image, task_type, predictor): # predictor as argument
     # IMPORTS INSIDE THIS FUNCTION TOO
     from diffusers.utils import export_to_video