Update app.py

app.py

@@ -33,15 +33,8 @@ torch._dynamo.config.recompile_limit = 1000
 torch._dynamo.config.capture_dynamic_output_shape_ops = True
 
 # Duration function
-def get_duration(prompt,seed,steps, duration_seconds,progress):
-                 
-    if duration_seconds <= 2.5
-        return steps * 18
-    else:
-        return steps * 25
 
-
-@spaces.GPU(duration=get_duration)
+@spaces.GPU(duration=60)
 def generate_video(prompt, seed, steps, duration_seconds,progress=gr.Progress(track_tqdm=True)):
     generator = torch.manual_seed(seed) if seed else None
     fps = 8