Spaces:

nithinraok
/

NeMo-Offline-Speaker-Diarization

Runtime error

nithinraok commited on Apr 28, 2023

Commit

9b7eef6

•

1 Parent(s): 0c7df68

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 model = NeuralDiarizer.from_pretrained("diar_msdd_telephonic").to(device)
 def run_diarization(path1):
-    annotation = model(path1, num_workers=0)
     rttm=annotation.to_rttm()
     df = pd.DataFrame(columns=['start_time', 'end_time', 'speaker'])
     for idx,line in enumerate(rttm.splitlines()):

 model = NeuralDiarizer.from_pretrained("diar_msdd_telephonic").to(device)
 def run_diarization(path1):
+    annotation = model(path1, num_workers=0, batch_size=16)
     rttm=annotation.to_rttm()
     df = pd.DataFrame(columns=['start_time', 'end_time', 'speaker'])
     for idx,line in enumerate(rttm.splitlines()):