try out libvpx-vp9

app.py

@@ -29,8 +29,7 @@ with gr.Blocks() as demo:
         btn = gr.Button("Generate")
         with gr.Row():
             with gr.Tab(label="Diffusion model"):
-                # video = gr.Video(label="Diffusion steps & Generated X-ray")
-                gr.HTML("<video preload='true' autoplay='true'><source src='/file/result.webm' type='video/webm'></video>")
+                video = gr.Video(label="Diffusion steps & Generated X-ray")
             with gr.Tab(label="GradCAM Overlay"):                
                 gr_image = gr.Image()
             with gr.Column():
@@ -41,12 +40,11 @@ with gr.Blocks() as demo:
     def submit(radio_selection):
         generated_video, generated_gradcam = generate_video_and_gradcam(radio_selection)
         return {
-            # video: generated_video,
+            video: generated_video,
             gr_image: generated_gradcam,
             disease_description: display_name(radio_selection)
         }
 
-    btn.click(fn=submit, inputs=radio, outputs=[gr_image, disease_description])
-    # btn.click(fn=submit, inputs=radio, outputs=[video, gr_image, disease_description])
+    btn.click(fn=submit, inputs=radio, outputs=[video, gr_image, disease_description])
 
 demo.launch()

create_video.py

@@ -30,5 +30,5 @@ def generate_video(numpy_image):
     for i, img in enumerate(image_lst):
         Image.fromarray(img).save(f"video/{i:03d}.jpg", quality=95)
 
-    ffmpeg.input('video/*.jpg', pattern_type='glob').output(save_path).run(overwrite_output=True)
+    ffmpeg.input('video/*.jpg', pattern_type='glob').output(save_path, vcodec="libvpx-vp9").run(overwrite_output=True)
     return save_path