use blocks api

app.py

@@ -1,10 +1,3 @@
-import os
-
-os.system("pip install gradio --upgrade")
-os.system("pip install torch")
-os.system("pip freeze")
-
-
 import torch
 import gradio as gr
 
@@ -33,18 +26,23 @@ def inference(video):
  return "com.mp4"
 
 
-title = "Robust Video Matting"
-description = "Gradio demo for Robust Video Matting. To use it, simply upload your video, or click one of the examples to load them. Read more at the links below."
-
-article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2108.11515'>Robust High-Resolution Video Matting with Temporal Guidance</a> | <a href='https://github.com/PeterL1n/RobustVideoMatting'>Github Repo</a></p>"
+with gr.Blocks(title="Robust Video Matting") as block:
+ gr.Markdown("# Robust Video Matting")
+ gr.Markdown(
+ "Gradio demo for Robust Video Matting. To use it, simply upload your video, or click one of the examples to load them. Read more at the links below."
+ )
+ with gr.Row():
+ inp = gr.Video(label="Input Video")
+ out = gr.Video(label="Output Video")
+ btn = gr.Button("Run")
+ btn.click(inference, inputs=inp, outputs=out)
+
+ gr.Examples(
+ examples=[["example.mp4"]],
+ inputs=[inp],
+ )
+ gr.HTML(
+ "<p style='text-align: center'><a href='https://arxiv.org/abs/2108.11515'>Robust High-Resolution Video Matting with Temporal Guidance</a> | <a href='https://github.com/PeterL1n/RobustVideoMatting'>Github Repo</a></p>"
+ )
 
-examples = [["example.mp4"]]
-gr.Interface(
- inference,
- gr.inputs.Video(label="Input"),
- gr.outputs.Video(label="Output"),
- title=title,
- description=description,
- article=article,
- examples=examples,
-).launch(debug=True)
+block.queue(api_open=False, max_size=5).launch(debug=True)

requirements.txt

@@ -4,4 +4,4 @@ torchaudio==2.1.0
 torchvision==0.16.0
 tqdm==4.61.1
 pims
-gradio
+gradio==3.16