height in playable and kptthresh

main.py

@@ -77,14 +77,14 @@ def show_tracking(video_content, vis_out_dir, model):
         return out_file
 
 
-def poses(inferencer, video, vis_out_dir):
+def poses(inferencer, video, vis_out_dir, kpt_thr):
 
     result_generator = inferencer(video, 
                                  vis_out_dir = vis_out_dir,
                                  return_vis=True,
                                  thickness=2,
                                  rebase_keypoint_height=True,
-                                 kpt_thr=0.15,
+                                 kpt_thr=kpt_thr,
                                  device="cuda"
                                  )    
     
@@ -94,7 +94,7 @@ def poses(inferencer, video, vis_out_dir):
 
     return out_file
 
-def infer(video, check):
+def infer(video, check, kpt_thr):
     # Selecting the specific inferencer
     out_files=[]
 
@@ -108,7 +108,7 @@ def infer(video, check):
             trackfile = show_tracking(video, vis_out_dir, inferencer)
 
         else:
-            out_file = poses(inferencer, video, vis_out_dir)                                                                                                                                                                                                    
+            out_file = poses(inferencer, video, vis_out_dir, kpt_thr)                                                                                                                                                                                                    
         
             out_files.extend(out_file)
         
@@ -122,20 +122,24 @@ def run():
     check_file = gr.CheckboxGroup(choices = ["Detect and track", "Estimate human 2d poses", "Estimate human 2d hand poses", "Estimate human 3d poses"], label="Methods", type="value", info="Select the model(s) you want")
     
     # Insert slider with kpt_thr
-
+    web_kpthr = gr.Slider(0, 1, value=0.3)
+    file_kpthr = gr.Slider(0, 1, value=0.3)
+    
     webcam = gr.Interface(
         fn=infer,
-        inputs= [gr.Video(source="webcam", height=512), check_web],
-        outputs = [gr.Video(format='mp4', height=512), gr.PlayableVideo(height=512), gr.PlayableVideo(height=512), gr.PlayableVideo(height=512)],
-        title = 'Pose estimation', 
-        description = 'Pose estimation on video',
+        inputs= [gr.Video(source="webcam", height=512), check_web, web_kpthr],
+        outputs = [gr.Video(format='mp4', height=512, label="Detect and track", show_label=True), gr.PlayableVideo(height=512,  label = "Estimate human 2d poses", show_label=True), gr.PlayableVideo(height=512, label = "Estimate human 2d hand poses", show_label=True), gr.PlayableVideo(height=512, label = "Estimate human 3d poses", show_label=True)],
+        title = 'Tracking and pose estimation', 
+        description = 'Tracking and pose estimation',
         allow_flagging=False
         )
 
     file = gr.Interface(
         infer,
-        inputs = [gr.Video(source="upload", height=512), check_file],
-        outputs = [gr.Video(format='mp4', height=512), gr.PlayableVideo(height=512), gr.PlayableVideo(height=512), gr.PlayableVideo(height=512)],
+        inputs = [gr.Video(source="upload", height=512), check_file, file_kpthr],
+        outputs = [gr.Video(format='mp4', height=512, label="Detect and track", show_label=True), gr.PlayableVideo(height=512,  label = "Estimate human 2d poses", show_label=True), gr.PlayableVideo(height=512, label = "Estimate human 2d hand poses", show_label=True), gr.PlayableVideo(height=512, label = "Estimate human 3d poses", show_label=True)],
+        title = 'Tracking and pose estimation', 
+        description = 'Tracking and pose estimation',
         allow_flagging=False
     )