taejinp commited on
Commit
49ffb84
·
verified ·
1 Parent(s): 8bffa83

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -234,7 +234,7 @@ The model is available for use in the NeMo Framework[6], and can be used as a pr
234
  from nemo.collections.asr.models import SortformerEncLabelModel
235
 
236
  # load model from Hugging Face model card directly (You need a Hugging Face token)
237
- diar_model = SortformerEncLabelModel.from_pretrained("nvidia/diar_sortformer_4spk-v1")
238
 
239
  # If you have a downloaded model in "/path/to/diar_streaming_sortformer_4spk-v2.nemo", load model from a downloaded file
240
  diar_model = SortformerEncLabelModel.restore_from(restore_path="/path/to/diar_streaming_sortformer_4spk-v2.nemo", map_location='cuda', strict=False)
 
234
  from nemo.collections.asr.models import SortformerEncLabelModel
235
 
236
  # load model from Hugging Face model card directly (You need a Hugging Face token)
237
+ diar_model = SortformerEncLabelModel.from_pretrained("nvidia/diar_streaming_sortformer_4spk-v2")
238
 
239
  # If you have a downloaded model in "/path/to/diar_streaming_sortformer_4spk-v2.nemo", load model from a downloaded file
240
  diar_model = SortformerEncLabelModel.restore_from(restore_path="/path/to/diar_streaming_sortformer_4spk-v2.nemo", map_location='cuda', strict=False)