Update sonogram_utility.py

sonogram_utility.py

@@ -140,6 +140,23 @@ def loadAudioTXT(sampleTXT):
 
     return [], prediction
 
+def loadAudioCSV(sampleCSV):
+    # Read in prediction data
+    df = pd.read_csv(sampleCSV)
+    
+    df = df.reset_index()  # make sure indexes pair with number of rows
+    
+    # Data in Annotation form, for convenient error rate calculation
+    prediction = Annotation(uri=sampleRTTM)
+    
+    for i, row in df.iterrows():
+        index = row['Resource']
+        start = (row['Start'] - row['Start'].date()).total_seconds()
+        end = (row['Finish'] - row['Finish'].date()).total_seconds()
+        prediction[Segment(start,end)] = index
+
+    return [], prediction
+    
 def splitIntoTimeSegments(testFile,maxDurationInSeconds=60):
 
     waveform, sample_rate = torchaudio.load(testFile)