Update README.md
Browse files
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/
|
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)
|