Audio Classification
speechbrain
PyTorch
English
Emotion
Diarization
wavlm
yingzhi commited on
Commit
c6f13d8
1 Parent(s): 9844a70

Update custom_interface.py

Browse files
Files changed (1) hide show
  1. custom_interface.py +0 -1
custom_interface.py CHANGED
@@ -128,7 +128,6 @@ class Speech_Emotion_Diarization(Pretrained):
128
  lol.append([batch_id[i], start, end, pred[j]])
129
 
130
  lol = merge_ssegs_same_emotion_adjacent(lol)
131
- print(lol)
132
  results[batch_id[i]] = [{"start": k[1], "end":k[2], "emotion": k[3]} for k in lol]
133
  return results
134
 
 
128
  lol.append([batch_id[i], start, end, pred[j]])
129
 
130
  lol = merge_ssegs_same_emotion_adjacent(lol)
 
131
  results[batch_id[i]] = [{"start": k[1], "end":k[2], "emotion": k[3]} for k in lol]
132
  return results
133