Update app_14B.py

app_14B.py

@@ -10,6 +10,7 @@ from diffusers import WanImageToVideoPipeline
 from diffusers.utils import export_to_video
 from huggingface_hub import upload_file
 from PIL import Image
+import spaces
 
 # ----------------- Setup -----------------
 logging.basicConfig(level=logging.INFO)
@@ -87,6 +88,21 @@ def upscale_and_upload_4k(input_video_path: str, input_image, summary_text: str)
     return hf_folder
 
 # ----------------- Video generation -----------------
+def get_duration(
+    input_image,
+    prompt,
+    negative_prompt,
+    duration_seconds,
+    guidance_scale,
+    guidance_scale_2,
+    steps,
+    seed,
+    randomize_seed,
+    progress,
+):
+    return steps * 15
+
+@spaces.GPU(duration=70)
 def generate_video(input_image, prompt, negative_prompt=default_negative_prompt,
                    duration_seconds=2, guidance_scale=3.5, steps=40, seed=0):
     if input_image is None: