Update app.py

app.py

@@ -51,12 +51,14 @@ def demo():
     with gr.Row():
         with gr.Column():
             image = gr.Image(label="Upload your image", type="pil")
+            resize_btn = gr.Button("Resize image")
+            resized_image = gr.Image(label="Resized image")
             with gr.Accordion("Advanced options", open=False):
                 seed = gr.Slider(label="Seed", value=42, randomize=True, minimum=0, maximum=max_64_bit_int, step=1)
                 randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
                 motion_bucket_id = gr.Slider(label="Motion bucket id", info="Controls how much motion to add/remove from the image", value=127, minimum=1, maximum=255)
                 fps_id = gr.Slider(label="Frames per second", info="The length of your video in seconds will be 25/fps", value=6, minimum=5, maximum=30)
-            generate_btn = gr.Button(value="Animate", variant="primary")
+            generate_btn = gr.Button("Animate")
         with gr.Column():
             video = gr.Video(label="Generated video")
             gallery = gr.Gallery(label="Generated frames")
@@ -64,10 +66,13 @@ def demo():
             progress = gr.Progress()
 
 
-    def resize_and_animate(image, progress, seed, randomize_seed, motion_bucket_id, fps_id):
-        image = resize_image(image)
-        if image.mode == "RGBA":
-            image = image.convert("RGB")
+    def resize(image):
+        return resize_image(image)
+
+
+    def animate(resized_image, progress, seed, randomize_seed, motion_bucket_id, fps_id):
+        if resized_image.mode == "RGBA":
+            resized_image = resized_image.convert("RGB")
             
         if(randomize_seed):
             seed = random.randint(0, max_64_bit_int)
@@ -79,7 +84,7 @@ def demo():
 
         frames = []
         for i in range(25):
-            frame = pipe(image, decode_chunk_size=3, generator=generator, motion_bucket_id=motion_bucket_id, noise_aug_strength=0.1, num_frames=1).frames[0]
+            frame = pipe(resized_image, decode_chunk_size=3, generator=generator, motion_bucket_id=motion_bucket_id, noise_aug_strength=0.1, num_frames=1).frames[0]
             frames.extend(frame)
             progress.update((i+1)/25)
 
@@ -89,10 +94,14 @@ def demo():
         return video_path, frames, seed
 
 
-    image.upload(fn=resize_image, inputs=image, outputs=image, queue=False)
+    resize_btn.click(
+        fn=resize,
+        inputs=image,
+        outputs=resized_image,
+    )
     generate_btn.click(
-        fn=resize_and_animate,
-        inputs=[image, progress, seed, randomize_seed, motion_bucket_id, fps_id],
+        fn=animate,
+        inputs=[resized_image, progress, seed, randomize_seed, motion_bucket_id, fps_id],
         outputs=[video, gallery, progress],
     )