On GPU

app.py

@@ -180,7 +180,6 @@ def resize_and_crop_to_match(target_image, reference_image):
     left, top = (new_width - ref_width) // 2, (new_height - ref_height) // 2
     return resized.crop((left, top, left + ref_width, top + ref_height))
 
-@spaces.GPU(duration=120)
 def generate_video(
     start_image_pil,
     end_image_pil,
@@ -193,6 +192,34 @@ def generate_video(
     seed=42,
     randomize_seed=False,
     progress=gr.Progress(track_tqdm=True)
+):
+    return generate_video(
+    start_image_pil,
+    end_image_pil,
+    prompt,
+    negative_prompt=default_negative_prompt,
+    duration_seconds,
+    steps,
+    guidance_scale,
+    guidance_scale_2,
+    seed,
+    randomize_seed,
+    progress
+)
+
+@spaces.GPU(duration=120)
+def generate_video_on_gpu(
+    start_image_pil,
+    end_image_pil,
+    prompt,
+    negative_prompt=default_negative_prompt,
+    duration_seconds=2.1,
+    steps=8,
+    guidance_scale=1,
+    guidance_scale_2=1,
+    seed=42,
+    randomize_seed=False,
+    progress=gr.Progress(track_tqdm=True)
 ):
     """
     Generates a video by interpolating between a start and end image, guided by a text prompt,
@@ -302,7 +329,7 @@ with gr.Blocks(theme=gr.themes.Citrus(), css=css) as app:
                 generate_button = gr.Button("Generate Video", variant="primary")
 
         with gr.Column():
-            output_video = gr.Video(label="Generated Video", autoplay=True)
+            output_video = gr.Video(label="Generated Video", autoplay = True, loop = True)
 
     # Main video generation button
     ui_inputs = [
@@ -378,7 +405,9 @@ with gr.Blocks(theme=gr.themes.Citrus(), css=css) as app:
         outputs=[]
     )
     
-    gr.Examples(
+    with gr.Row(elem_id="image_examples", visible=False):
+        gr.Examples(
+        label = "Examples from images",
         examples = [
             ["poli_tower.png", "tower_takes_off.png", "the man turns around"],
             ["ugly_sonic.jpeg", "squatting_sonic.png", "the character dodges the missiles"],
@@ -390,6 +419,18 @@ with gr.Blocks(theme=gr.themes.Citrus(), css=css) as app:
         run_on_click = True,
         cache_examples = True,
     )
+    
+    gr.Examples(
+        examples = [
+            ["poli_tower.png", "tower_takes_off.png", "the man turns around"],
+            ["ugly_sonic.jpeg", "squatting_sonic.png", "the character dodges the missiles"],
+            ["capyabara_zoomed.png", "capyabara.webp", "a dramatic dolly zoom"],
+        ],
+        inputs = [start_image, end_image, prompt],
+        outputs = ui_outputs,
+        fn = generate_video,
+        cache_examples = False,
+    )
 
 if __name__ == "__main__":
     app.launch(share=True)