Convert all labels to string IDs

sonogram.py

@@ -129,6 +129,18 @@ class Sonogram():
             diarization[segment] = 'group'
             
         totalTimeInSeconds = int(waveform.shape[-1]/sample_rate)
+
+        # Rename labels
+        currId = 0
+        mapping = {}
+        for label in diarization.lables():
+            if label == 'group':
+                continue
+            else:
+                currId += 1
+                newLabel = f'SPEAKER_{currId:03d}'
+                mapping[label] = newLabel
+        diarization = diarization.rename_lables(mapping)
         print("Time in seconds calculated")
         return diarization, totalTimeInSeconds, waveform, sample_rate