Update app.py

app.py

@@ -19,6 +19,7 @@ DEFAULT_MODEL_ID = os.getenv(
     "HF_ZERO_GPU_VIDEO_MODEL",
     "stabilityai/stable-video-diffusion-img2vid-xt-1-1",
 )
+DEFAULT_GPU_DURATION_SECONDS = int(os.getenv("HF_ZERO_GPU_DURATION_SECONDS", "600"))
 DEFAULT_FPS = int(os.getenv("HF_ZERO_GPU_FPS", "7"))
 DEFAULT_NUM_FRAMES = int(os.getenv("HF_ZERO_GPU_NUM_FRAMES", "25"))
 DEFAULT_INFERENCE_STEPS = int(os.getenv("HF_ZERO_GPU_INFERENCE_STEPS", "25"))
@@ -95,7 +96,7 @@ def _render_clip(
     export_to_video(frames, str(output_path), fps=fps)
 
 
-@spaces.GPU(duration=900)
+@spaces.GPU(duration=DEFAULT_GPU_DURATION_SECONDS)
 def render_package(
     package_zip: str,
     model_id: str,