file display

main.py

@@ -43,37 +43,6 @@ def tracking(video, model, boxes=True):
 
 
 
-#def show_tracking(video_content):
-
-        video = cv2.VideoCapture(video_content)
-
-        # Track
-        video_track = tracking(video_content, track_model.track)
-
-        # Prepare to save video
-        output_file_track = os.path.join("track.mp4")
-
-        fourcc = cv2.VideoWriter_fourcc(*"mp4v")  # Codec for MP4 video
-        fps = video.get(cv2.CAP_PROP_FPS)
-        height, width, _ = video_track[0][0].orig_img.shape
-        size = (width,height)
-
-        out_track = cv2.VideoWriter(output_file_track, fourcc, fps, size)
-
-        # Go through frames and write them 
-        for frame_track in video_track:
-            result_track = frame_track[0].plot()  # plot a BGR numpy array of predictions
-
-        #print(type(result_pose)) numpy ndarray
-        out_track.write(result_track)
-
-        out_track.release()
-
-        video.release()
-        cv2.destroyAllWindows() # Closing window
-
-        return output_file_track
-
 def poses(photo, check):
     # Selecting the specific inferencer
     out_files=[]
@@ -107,7 +76,7 @@ def run():
     webcam = gr.Interface(
         fn=poses,
         inputs= [gr.Video(source="webcam", height=412), check_web],
-        outputs = [gr.PlayableVideo(),gr.PlayableVideo(),gr.PlayableVideo()],  #file_types=['.mp4'] #gr.Video(),
+        outputs = [gr.PlayableVideo(), gr.PlayableVideo(), gr.PlayableVideo()],
         title = 'Pose estimation', 
         description = 'Pose estimation on video',
         allow_flagging=False