stooop

main_noweb.py

@@ -66,7 +66,6 @@ def check_extension(video):
 
 
 def pose3d(video, kpt_threshold):
-
     video = check_extension(video)
     print(device)
     
@@ -209,7 +208,7 @@ with block:
                 with gr.Row():
                     with gr.Column():
                         with gr.Row():
-                            video_input_web = gr.Video(source="webcam", height=256, width=192)
+                            video_input_web = gr.Video(source="webcam", include_audio=False, height=256, width=192)
                             # Insert slider with kpt_thr
                             with gr.Column():
                                 gr.Markdown("Drag the keypoint threshold to filter out lower probability keypoints:")
@@ -218,6 +217,8 @@ with block:
                             submit_pose_file_web = gr.Button("Make 2d pose estimation")
                             submit_pose3d_file_web = gr.Button("Make 3d pose estimation")
                             submit_hand_file_web = gr.Button("Make 2d hand estimation")
+                        stop = gr.Button(value="Stop estimation")
+
 
                 with gr.Row():
                     video_output1_web = gr.PlayableVideo(label = "Estimate human 2d poses", show_label=True, height=256)
@@ -241,7 +242,7 @@ with block:
                 with gr.Row():
                     with gr.Column():
                         with gr.Row():
-                            video_input = gr.Video(source="upload", type="filepath", height=256, width=192)
+                            video_input = gr.Video(source="upload", type="filepath", include_audio=False, height=256, width=192)
                             # Insert slider with kpt_thr
                             with gr.Column():
                                 gr.Markdown("Drag the keypoint threshold to filter out lower probability keypoints:")
@@ -324,23 +325,26 @@ with block:
             
 
         # From file
-        submit_pose_file.click(fn=pose2d, 
+        click1 = submit_pose_file.click(fn=pose2d, 
                             inputs=  [video_input, file_kpthr], 
                             outputs = [video_output1, jsonoutput],
                             queue=True)
         
-        submit_pose3d_file.click(fn=pose3d, 
+        click2 = submit_pose3d_file.click(fn=pose3d, 
                                 inputs= [video_input, file_kpthr], 
                                 outputs = [video_output2, jsonoutput],
                                 #batch=True, 
                                 #max_batch_size=16,
                                 queue=True) # Sometimes it worked with queue false? But still slow
         
-        submit_hand_file.click(fn=pose2dhand, 
+        click3 = submit_hand_file.click(fn=pose2dhand, 
                             inputs= [video_input, file_kpthr], 
                             outputs = [video_output3, jsonoutput],
                             queue=True)
         
+        stop.click(fn=None, inputs=None, outputs=None, cancels=[click1, click2, click3])
+
+
         # From web
         submit_pose_file_web.click(fn=pose2d, 
                             inputs=  [video_input_web, file_kpthr_web],