added return in check extension

main_noweb.py

@@ -101,8 +101,6 @@ def pose3d(video, kpt_threshold):
     print(device)
     
     human3d = MMPoseInferencer(device=device,   pose3d="human3d",  scope="mmpose")#"pose-lift_videopose3d-243frm-supv-cpn-ft_8xb128-200e_h36m")
-
-    print("HUMAN 3d downloaded!!")    
     # Define new unique folder
     add_dir = str(uuid.uuid4())
 
@@ -117,7 +115,6 @@ def pose3d(video, kpt_threshold):
                             kpt_thr=kpt_threshold,
                             pred_out_dir = add_dir
                             )
-    print("INFERENCE DONW")
     result = [result for result in result_generator] #next(result_generator)        
 
     out_file = glob.glob(os.path.join(add_dir, "*.mp4")) #+ glob.glob(os.path.join(vis_out_dir, "*.webm")) 
@@ -129,8 +126,8 @@ def pose3d(video, kpt_threshold):
 
 
 def pose2d(video, kpt_threshold):
-    #video = check_extension(video)
-    video = get_frames(video)
+    video = check_extension(video)
+    #video = get_frames(video)
 
     # Define new unique folder
     add_dir = str(uuid.uuid4())
@@ -226,7 +223,7 @@ block = gr.Blocks()
 
 with block:
     with gr.Column():       
-        with gr.Tab("Upload video"):
+        with gr.Tab("Capture video with webcam"):
             with gr.Column():
                 with gr.Row():
                     with gr.Column():
@@ -326,12 +323,12 @@ with block:
                         13: Left Knee
                         14: Right Knee
                         15: Left Ankle
-                        16: Right Ankle 
-                        ```
+                        16: Right Ankle ```
 
                         \n Below, you can see a visualization of the poses of the 2d, 3d and hand keypoint locations: """)
-            gr.Image("./cocoposes.png", width="200")
-            gr.Image("./cocohand.png", width="200")
+            
+            gr.Image("./cocoposes.png", width="160")
+            gr.Image("./cocohand.png", width="160")