Update visualization.py

visualization.py

@@ -314,6 +314,9 @@ def create_video_with_heatmap(video_path, df, mse_embeddings, mse_posture, mse_v
                 progress_percent = float(msg.split(' ')[-1].replace('%', '')) / 100
                 self.progress_callback(progress_percent, msg)
 
+        def __call__(self, msg, *args, **kwargs):
+            self.debug(msg, *args, **kwargs)
+
     logging.basicConfig(level=logging.DEBUG)
     progress_logger = ProgressBarLogger('progress_logger', progress)