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)
|