Update app.py

app.py

@@ -62,19 +62,19 @@ default_prompt_t2v = "Two anthropomorphic cats in comfy boxing gear and bright g
 default_negative_prompt = "色调艳丽, 过曝, 静态, 细节模糊不清, 字幕, 风格, 作品, 画作, 画面, 静止, 整体发灰, 最差质量, 低质量, JPEG压缩残留, 丑陋的, 残缺的, 多余的手指, 画得不好的手部, 画得不好的脸部, 畸形的, 毁容的, 形态畸形的肢体, 手指融合, 静止不动的画面, 杂乱的背景, 三条腿, 背景人很多, 倒着走"
 
 
-# def get_duration(
-#     prompt,
-#     negative_prompt,
-#     guidance_scale,
-#     guidance_scale_2,
-#     steps,
-#     seed,
-#     randomize_seed,
-#     progress,
-# ):
-#     return steps * 15
-
-@spaces.GPU()
+def get_duration(
+    prompt,
+    negative_prompt,
+    guidance_scale,
+    guidance_scale_2,
+    steps,
+    seed,
+    randomize_seed,
+    progress,
+):
+    return steps * 15
+
+@spaces.GPU(duration = get_duration)
 def generate_image(
     prompt,
     negative_prompt=default_negative_prompt,