Update app.py

app.py

@@ -149,7 +149,7 @@ def initialize_models():
     if latent_upsampler_instance:
         latent_upsampler_instance.to(target_inference_device)
 
-@spaces.GPU(duration=45)
+@spaces.GPU(duration=60)
 def generate(prompt, input_image_url=None, final_image_url=None, duration_ui=2, progress=gr.Progress(track_tqdm=True)):
     """Generate video from image(s) and prompt"""
     # Validate input - at least one image must be provided
@@ -331,12 +331,7 @@ def generate(prompt, input_image_url=None, final_image_url=None, duration_ui=2,
 
     if result_images_tensor is None:
         raise gr.Error("Generation failed.")
-    else:
-        return result_images_tensor
-
 
-def generate2(prompt, input_image_url=None, final_image_url=None, duration_ui=2, progress=gr.Progress(track_tqdm=True)):
-    result_images_tensor = generate(prompt, input_image_url, final_image_url, duration_ui, progress)
     # Process the result
     pad_left, pad_right, pad_top, pad_bottom = padding_values
     slice_h_end = -pad_bottom if pad_bottom > 0 else None
@@ -411,7 +406,7 @@ with gr.Blocks(css=css) as demo:
 
     gr.Markdown("**Note:** You must provide at least one input image (either first frame or last frame).")
     generate_button.click(
-        fn=generate2,
+        fn=generate,
         inputs=[prompt_input, input_image_input, final_image_input, duration_input],
         outputs=[video_output],
         api_name="generate_video"