blocks

main_blocks.py

@@ -44,11 +44,11 @@ def tracking(video, model, boxes=True):
 
     return annotated_frame
 
-def show_tracking(video_content, vis_out_dir, model):
+def show_tracking(video_content):
         video = cv2.VideoCapture(video_content)
 
         # Track
-        video_track = tracking(video_content, model.track)
+        video_track = tracking(video_content, track_model.track)
 
         # Prepare to save video
         #out_file = os.path.join(vis_out_dir, "track.mp4")