update batch infer

modeling_finelap.py

@@ -154,7 +154,7 @@ class FineLAPModel(PreTrainedModel):
         import matplotlib.pyplot as plt
         import numpy as np
 
-        scores = self.get_frame_level_score(audio_path, text_labels, device)
+        scores = self.get_frame_level_score(audio_path, text_labels, device).squeeze()
         sim_matrix_np = scores.cpu().numpy()
 
         fig, ax = plt.subplots(figsize=(14, 8))