Update visualization.py

visualization.py

@@ -310,9 +310,11 @@ def create_video_with_heatmap(video_path, df, mse_embeddings, mse_posture, mse_v
             self.progress_callback = progress_callback
 
         def debug(self, msg, *args, **kwargs):
-            if self.progress_callback:
-                self.progress_callback(float(msg.split('%')[0]) / 100, msg)
+            if 't=' in msg and self.progress_callback:
+                progress_percent = float(msg.split(' ')[-1].replace('%', '')) / 100
+                self.progress_callback(progress_percent, msg)
 
+    logging.basicConfig(level=logging.DEBUG)
     progress_logger = ProgressBarLogger('progress_logger', progress)
 
     # Write the final video