Update README.md

README.md

@@ -5,7 +5,6 @@ license: apache-2.0
 
 This is a copy of `voxceleb_resnet34_LM.onnx` speaker embedding model taken from [ChengDong Liang's repository](https://huggingface.co/chengdongliang/wespeaker).
 
-
 ```
 Copyright 2022 ChengDong Liang
 
@@ -20,4 +19,17 @@ distributed under the License is distributed on an "AS IS" BASIS,
 WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 See the License for the specific language governing permissions and
 limitations under the License.
-```
+```
+
+```python
+from pyannote.audio.pipelines.speaker_verification import PretrainedSpeakerEmbedding
+get_embedding = PretrainedSpeakerEmbedding("hbredin/wespeaker-voxceleb-resnet34-LM")
+
+assert waveforms.ndim == 3
+batch_size, num_channels, num_samples = waveforms.shape
+assert num_channels == 1
+
+embeddings = get_embedding(waveforms)
+assert embeddings.ndim == 2
+assert embeddings.shape[0] == batch_size
+```