xmrt commited on
Commit
e973eb8
·
1 Parent(s): 0e929df
Files changed (1) hide show
  1. main.py +5 -3
main.py CHANGED
@@ -88,10 +88,11 @@ def poses(photo, check):
88
  def run():
89
  #https://github.com/open-mmlab/mmpose/blob/main/docs/en/user_guides/inference.md
90
  available_methods = ["Estimate human 2d poses", "Estimate human 2d hand poses", "Estimate human 3d poses"]
91
- check = gr.CheckboxGroup(available_methods, label="Methods", type="value", info="Select the model(s) you want")
92
  webcam = gr.Interface(
93
  fn=poses,
94
- inputs= [gr.Video(source="webcam"), check],
 
95
  outputs = gr.Video(format='mp4'),
96
  title = 'Pose estimation',
97
  description = 'Pose estimation on video',
@@ -100,7 +101,8 @@ def run():
100
 
101
  file = gr.Interface(
102
  poses,
103
- inputs = [gr.Video(source="upload"), check],
 
104
  outputs = gr.Video(format='mp4'),
105
  allow_flagging=False
106
  )
 
88
  def run():
89
  #https://github.com/open-mmlab/mmpose/blob/main/docs/en/user_guides/inference.md
90
  available_methods = ["Estimate human 2d poses", "Estimate human 2d hand poses", "Estimate human 3d poses"]
91
+
92
  webcam = gr.Interface(
93
  fn=poses,
94
+ inputs= [gr.Video(source="webcam"),
95
+ gr.CheckboxGroup(available_methods, label="Methods", type="value", info="Select the model(s) you want")],
96
  outputs = gr.Video(format='mp4'),
97
  title = 'Pose estimation',
98
  description = 'Pose estimation on video',
 
101
 
102
  file = gr.Interface(
103
  poses,
104
+ inputs = [gr.Video(source="upload"),
105
+ gr.CheckboxGroup(available_methods, label="Methods", type="value", info="Select the model(s) you want")],
106
  outputs = gr.Video(format='mp4'),
107
  allow_flagging=False
108
  )