Update app.py

app.py

@@ -15,7 +15,8 @@ def run_diarization(path1):
     df = pd.DataFrame(columns=['start_time', 'end_time', 'speaker'])
     for idx,line in enumerate(rttm.splitlines()):
         split = line.split()
-        start_time, end_time, speaker = split[3], split[4], split[7]
+        start_time, duration, speaker = split[3], split[4], split[7]
+        end_time = float(start_time) + float(duration)
         df.loc[idx] = start_time, end_time, speaker
     return df 
 	
@@ -37,7 +38,8 @@ article = (
     "</p>"
 )
 examples = [
-    "data/id10270_5r0dWxy17C8-00001.wav",
+    ["data/sample_interview_conversation.wav"],
+    ["data/id10270_5r0dWxy17C8-00001.wav"],
 ]
 
 interface = gr.Interface(