Update app.py

app.py

@@ -57,8 +57,18 @@ def sample(
     frames = pipe(image, decode_chunk_size=decoding_t, generator=generator, motion_bucket_id=motion_bucket_id, noise_aug_strength=0.1, num_frames=25).frames[0]
     export_to_video(frames, video_path, fps=fps_id)
     torch.manual_seed(seed)
+
+    # Read the content of the video file and encode it to base64
+    with open(video_path, "rb") as video_file:
+        video_base64 = base64.b64encode(video_file.read()).decode('utf-8')
+
+    # Prepend the appropriate data URI header with MIME type
+    video_data_uri = 'data:video/mp4;base64,' + video_base64
     
-    return video_path, seed
+    # Clean up the video file to avoid filling up storage
+    # os.remove(video_path)
+
+    return video_data_uri
 
 def resize_image(image, output_size=(1024, 576)):
     # Calculate aspect ratios
@@ -104,13 +114,14 @@ with gr.Blocks() as demo:
     with gr.Column():
         image = gr.Image(label="Upload your image", type="pil")
         generate_btn = gr.Button("Generate")
-    video = gr.Video()
+    base64_out = gr.Textbox(label="Base64 Video")
+
   with gr.Accordion("Advanced options", open=False):
       seed = gr.Slider(label="Seed", value=42, randomize=False, minimum=0, maximum=max_64_bit_int, step=1)
       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.click(fn=sample, inputs=[image, seed, motion_bucket_id, fps_id, secret_token], outputs=video, api_name="video")
+  generate_btn.click(fn=sample, inputs=[image, seed, motion_bucket_id, fps_id, secret_token], outputs=[base64_out], api_name="video")
 
 if __name__ == "__main__":
     demo.queue(max_size=20).launch()