Update app.py

app.py

@@ -1,20 +1,20 @@
-# from nemo.collections.asr.models import NeuralDiarizer
+from nemo.collections.asr.models import NeuralDiarizer
 import gradio as gr
 import pandas as pd
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
-# model = NeuralDiarizer.from_pretrained("diar_msdd_telephonic").to(device)
+model = NeuralDiarizer.from_pretrained("diar_msdd_telephonic").to(device)
 
 def run_diarization(path1):
-    # annotation = model(path1)
-    # rttm=annotation.to_rttm()
+    annotation = model(path1)
+    rttm=annotation.to_rttm()
     df = pd.DataFrame(columns=['start_time', 'end_time', 'speaker'])
-    # for idx,line in enumerate(rttm.splitlines()):
-    #     split = line.split()
-    #     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
+    for idx,line in enumerate(rttm.splitlines()):
+        split = line.split()
+        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
 	
 inputs = [