port

main.py

@@ -88,6 +88,7 @@ def pose3d(video):
     result_generator = human3d(video, 
                                  vis_out_dir = vis_out_dir,
                                  thickness=2,
+                                 return_vis=True,
                                  rebase_keypoint_height=True,
                                  device="cuda")    
     
@@ -107,6 +108,7 @@ def pose2d(video):
     result_generator = human(video, 
                                  vis_out_dir = vis_out_dir,
                                  thickness=2,
+                                 return_vis=True,
                                  rebase_keypoint_height=True,
                                  device="cuda")    
     
@@ -128,6 +130,7 @@ def pose2dhand(video):
 
     result_generator = hand(video, 
                                  vis_out_dir = vis_out_dir,
+                                 return_vis=True,
                                  thickness=2,
                                  rebase_keypoint_height=True,
                                  device="cuda")    
@@ -148,7 +151,6 @@ with gr.Blocks() as demo:
                 with gr.Column():
                     video_input = gr.Video(source="upload", type="filepath", height=512)
                     # Insert slider with kpt_thr
-                    web_kpthr = gr.Slider(0, 1, value=0.3)
                     file_kpthr = gr.Slider(0, 1, value=0.3)
 
                     submit_pose_file = gr.Button("Make 2d pose estimation")
@@ -167,7 +169,6 @@ with gr.Blocks() as demo:
                     webcam_input = gr.Video(source="webcam", height=512)
                     
                     web_kpthr = gr.Slider(0, 1, value=0.3)
-                    file_kpthr = gr.Slider(0, 1, value=0.3)
 
                     submit_pose_web = gr.Button("Make 2d pose estimation")
                     submit_pose3d_web = gr.Button("Make 3d pose estimation")