Ahsen Khaliq commited on
Commit
3f9359c
·
1 Parent(s): 1a3c553

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -2
app.py CHANGED
@@ -6,7 +6,15 @@ def greet(vid):
6
  os.system("python demos/general.py --weights kapao_s_coco.pt --start 0 --end 2 --gif --fps --vid "+vid)
7
  return 'flash_mob_inference_kapao_s_coco.gif'
8
 
9
-
 
 
 
10
 
11
- iface = gr.Interface(fn=greet,inputs="video", outputs=gr.outputs.Image(type="file", label="output"), enable_queue=True)
 
 
 
 
 
12
  iface.launch()
 
6
  os.system("python demos/general.py --weights kapao_s_coco.pt --start 0 --end 2 --gif --fps --vid "+vid)
7
  return 'flash_mob_inference_kapao_s_coco.gif'
8
 
9
+ title = "Kapao"
10
+ description = "Gradio demo for Kapao: an efficient multi-person human pose estimation model that detects keypoints and poses as objects and fuses the detections to predict human poses. To use it, simply upload your video, or click one of the examples to load them. Read more at the links below."
11
+ article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2111.08557' target='_blank'>Rethinking Keypoint Representations: Modeling Keypoints and Poses as Objects for Multi-Person Human Pose Estimation</a> | <a href='https://github.com/wmcnally/kapao' target='_blank'>Github Repo</a></p>"
12
+
13
 
14
+ iface = gr.Interface(fn=greet,inputs="video", outputs=gr.outputs.Image(type="file", label="output"), enable_queue=True, title=title,
15
+ description=description,
16
+ article=article,
17
+ examples=[
18
+ ["example.mp4"]
19
+ ])
20
  iface.launch()