Spaces:

luckyhookin
/

speaker-diarization-diar_sortformer

Running on Zero

luckyhookin commited on 12 days ago

Commit

6d43bc6

1 Parent(s): e1e635d

update

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ import json
 import requests
 from nemo.collections.asr.models import SortformerEncLabelModel
-diar_model = SortformerEncLabelModel.from_pretrained("nvidia/diar_streaming_sortformer_4spk-v2")
 diar_model.eval()
 diar_model.sortformer_modules.chunk_len = 340
@@ -95,7 +95,7 @@ def format_results(results):
 # Gradio interface
 with gr.Blocks() as demo:
-    gr.Markdown("# Speaker Diarization with nvidia/diar_streaming_sortformer_4spk-v2")
     gr.Markdown("Upload an audio file and specify the number of speakers to diarize the audio.")
     with gr.Row():

 import requests
 from nemo.collections.asr.models import SortformerEncLabelModel
+diar_model = SortformerEncLabelModel.from_pretrained("nvidia/diar_streaming_sortformer_4spk-v2.1")
 diar_model.eval()
 diar_model.sortformer_modules.chunk_len = 340
 # Gradio interface
 with gr.Blocks() as demo:
+    gr.Markdown("# Speaker Diarization with nvidia/diar_streaming_sortformer_4spk-v2.1")
     gr.Markdown("Upload an audio file and specify the number of speakers to diarize the audio.")
     with gr.Row():