output

main.py

@@ -55,6 +55,7 @@ def check_extension(video):
 
 
 def tracking(video, model, boxes=True):
+    print("[INFO] Is cuda available? ", torch.cuda.is_available())
     print("[INFO] Loading model...")
     # Load an official or custom model
 
@@ -66,6 +67,8 @@ def tracking(video, model, boxes=True):
     return annotated_frame
 
 def show_tracking(video_content):
+        print()
+
         # https://docs.ultralytics.com/datasets/detect/coco/
         video = cv2.VideoCapture(video_content)
 
@@ -177,7 +180,7 @@ def run_UI():
                     with gr.Column():
                         video_input = gr.Video(source="upload", type="filepath", height=612)
                         # Insert slider with kpt_thr
-                        file_kpthr = gr.Slider(minimum=1e3, maximum=1e6, step=1e3, default=1e3, label='Keypoint threshold')
+                        file_kpthr = gr.Slider(minimum=0.1, maximum=1, step=1e3, default=0.3, label='Keypoint threshold')
 
                         submit_pose_file = gr.Button("Make 2d pose estimation")
                         submit_pose3d_file = gr.Button("Make 3d pose estimation")
@@ -191,7 +194,7 @@ def run_UI():
 
             with gr.Tab("Record video with webcam"):
                 
-                with gr.Column:
+                with gr.Column():
                     with gr.Row():
                         with gr.Column():
                             webcam_input = gr.Video(source="webcam", height=612)