taejinp commited on
Commit
bc6cc3e
·
verified ·
1 Parent(s): d2207a6

Updated license and model loading

Browse files
Files changed (1) hide show
  1. README.md +4 -2
README.md CHANGED
@@ -143,8 +143,10 @@ The model is available for use in the NeMo Framework[5], and can be used as a pr
143
  ```python
144
  from nemo.collections.asr.models import SortformerEncLabelModel
145
 
146
- # load model
147
  diar_model = SortformerEncLabelModel.restore_from(restore_path="/path/to/diar_sortformer_4spk-v1.nemo", map_location=torch.device('cuda'), strict=False)
 
 
148
  ```
149
 
150
  ### Input Format
@@ -322,4 +324,4 @@ Check out [Riva live demo](https://developer.nvidia.com/riva#demos).
322
 
323
  ## Licence
324
 
325
- License to use this model is covered by the [CC-BY-NC-SA-4.0](https://creativecommons.org/licenses/by-nc-sa/4.0/legalcode). By downloading the public and release version of the model, you accept the terms and conditions of the CC-BY-NC-SA-4.0 license.
 
143
  ```python
144
  from nemo.collections.asr.models import SortformerEncLabelModel
145
 
146
+ # load model from a downloaded file
147
  diar_model = SortformerEncLabelModel.restore_from(restore_path="/path/to/diar_sortformer_4spk-v1.nemo", map_location=torch.device('cuda'), strict=False)
148
+ # load model from Hugging Face model card directly (You need a Hugging Face token)
149
+ diar_model = SortformerEncLabelModel.from_pretrained("nvidia/diar_sortformer_4spk-v1")
150
  ```
151
 
152
  ### Input Format
 
324
 
325
  ## Licence
326
 
327
+ License to use this model is covered by the [CC-BY-NC-4.0](https://creativecommons.org/licenses/by-nc-sa/4.0/legalcode). By downloading the public and release version of the model, you accept the terms and conditions of the CC-BY-NC-SA-4.0 license.