skipping tracking

main.py

@@ -49,7 +49,7 @@ def show_tracking(video_content, vis_out_dir, model):
         video_track = tracking(video_content, model.track)
 
         # Prepare to save video
-        out_file = os.path.join(vis_out_dir, "track.mp4")
+        #out_file = os.path.join(vis_out_dir, "track.mp4")
         out_file = "track.mp4"
         print("[INFO]: TRACK", out_file)
 
@@ -100,7 +100,7 @@ def infer(video, check):
         inferencer = inferencers[i] # 'hand', 'human , device='cuda'
 
         if i == "Detect and track":
-            continue
+            #continue
             out_file = show_tracking(video, vis_out_dir, inferencer)
 
         else:
@@ -110,7 +110,7 @@ def infer(video, check):
         out_files.extend(out_file)
         print(out_files)
 
-    return out_files[0], out_files[1], out_files[2]#, out_files[3]
+    return out_files[0], out_files[1], out_files[2], out_file # out_files[3]
 
 def run():
     #https://github.com/open-mmlab/mmpose/blob/main/docs/en/user_guides/inference.md
@@ -120,7 +120,7 @@ def run():
     webcam = gr.Interface(
         fn=infer,
         inputs= [gr.Video(source="webcam", height=412), check_web],
-        outputs = [gr.PlayableVideo(), gr.PlayableVideo(), gr.PlayableVideo()],# gr.PlayableVideo()],
+        outputs = [gr.PlayableVideo(), gr.PlayableVideo(), gr.PlayableVideo() gr.PlayableVideo()],
         title = 'Pose estimation', 
         description = 'Pose estimation on video',
         allow_flagging=False
@@ -129,7 +129,7 @@ def run():
     file = gr.Interface(
         infer,
         inputs = [gr.Video(source="upload", height=412), check_file],
-        outputs = [gr.PlayableVideo(), gr.PlayableVideo(), gr.PlayableVideo()],#, gr.PlayableVideo()],
+        outputs = [gr.PlayableVideo(), gr.PlayableVideo(), gr.PlayableVideo(), gr.PlayableVideo()],
         allow_flagging=False
     )