xmrt commited on
Commit
bbdb0f2
1 Parent(s): a4769fb
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -156,7 +156,7 @@ def pose2d(video, kpt_threshold):
156
  out_file = glob.glob(os.path.join(vis_out_dir, "*.mp4")) #+ glob.glob(os.path.join(vis_out_dir, "*.webm"))
157
  kpoints = glob.glob(os.path.join(vis_out_dir, "*.json"))
158
 
159
- return "".join(out_file), kpoints
160
 
161
 
162
  def pose2dhand(video, kpt_threshold):
@@ -206,7 +206,7 @@ with block:
206
  video_output3 = gr.PlayableVideo(height=512, label = "Estimate human hand poses", show_label=True)
207
  video_output4 = gr.Video(height=512, label = "Detection and tracking", show_label=True, format="mp4")
208
  jsonoutput = gr.Code("json")
209
-
210
  with gr.Tab("Record video with webcam"):
211
 
212
  with gr.Column():
 
156
  out_file = glob.glob(os.path.join(vis_out_dir, "*.mp4")) #+ glob.glob(os.path.join(vis_out_dir, "*.webm"))
157
  kpoints = glob.glob(os.path.join(vis_out_dir, "*.json"))
158
 
159
+ return "".join(out_file), "".join(kpoints)
160
 
161
 
162
  def pose2dhand(video, kpt_threshold):
 
206
  video_output3 = gr.PlayableVideo(height=512, label = "Estimate human hand poses", show_label=True)
207
  video_output4 = gr.Video(height=512, label = "Detection and tracking", show_label=True, format="mp4")
208
  jsonoutput = gr.Code("json")
209
+
210
  with gr.Tab("Record video with webcam"):
211
 
212
  with gr.Column():