Update sonogram_utility.py

sonogram_utility.py

@@ -96,7 +96,7 @@ def loadAudioRTTM(sampleRTTM):
             while len(speakerList) < index + 1:
                 speakerList.append([])
             speakerList[index].append((float(speakerResult[3]),float(speakerResult[4])))
-            prediction[Segment(start,end)] = index
+            prediction[Segment(start,end)] = speakerResult[7]
 
     return speakerList, prediction