port

main.py

@@ -147,59 +147,72 @@ with gr.Blocks() as demo:
             with gr.Row():
                 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")
                     submit_pose3d_file = gr.Button("Make 3d pose estimation")
                     submit_hand_file = gr.Button("Make 2d hand estimation")
                     submit_detect_file = gr.Button("Detect and track objects")
 
-                video_output = gr.Video(height=512)
+                video_output1 = gr.Video(height=512)
+                video_output2 = gr.Video(height=512)
+                video_output3 = gr.Video(height=512)
+                video_output4 = gr.Video(height=512)
 
         with gr.Tab("Record video with webcam"):
             with gr.Row():
                 with gr.Column():
                     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")
                     submit_hand_web = gr.Button("Make 2d hand estimation")
                     submit_detect_web = gr.Button("Detect and track objects")
 
-                webcam_output = gr.Video(height=512)
+                webcam_output1 = gr.Video(height=512)
+                webcam_output2 = gr.Video(height=512)
+                webcam_output3 = gr.Video(height=512)
+                webcam_output4 = gr.Video(height=512)
+
         
     
     # From file
     submit_pose_file.click(fn=pose2d, 
                            inputs= video_input, 
-                           outputs = video_output)
+                           outputs = video_output1)
     
     submit_pose3d_file.click(fn=pose3d, 
                              inputs= video_input, 
-                             outputs = video_output)
+                             outputs = video_output2)
     
     submit_hand_file.click(fn=pose2dhand, 
                            inputs= video_input, 
-                           outputs = video_output)
+                           outputs = video_output3)
     
     submit_detect_file.click(fn=show_tracking, 
                              inputs= video_input, 
-                             outputs = video_output)
+                             outputs = video_output4)
     
     # Web
     submit_pose_web.click(fn=pose2d, 
-                          inputs= video_input, 
-                          outputs = video_output)
+                          inputs= webcam_input, 
+                          outputs = webcam_output1)
     
     submit_pose3d_web.click(fn=pose3d, 
-                            inputs= video_input, 
-                            outputs = video_output)
+                            inputs= webcam_input, 
+                            outputs = webcam_output2)
     
     submit_hand_web.click(fn=pose2dhand, 
-                          inputs= video_input, 
-                          outputs = video_output)
+                          inputs= webcam_input, 
+                          outputs = video_output3)
     
     submit_detect_web.click(fn=show_tracking, 
-                            inputs= video_input, 
-                            outputs = video_output)
+                            inputs= webcam_input, 
+                            outputs = video_output4)
 
 demo.launch(server_name="0.0.0.0", server_port=7860)