Update README.md

README.md

@@ -28,8 +28,50 @@ It achieves the following results on the evaluation set:
 - Confusion: 0.0520
 
 ## Model description
+This segmentation model has been trained on English data (backup_uganda) using [diarizers](https://github.com/huggingface/diarizers/tree/main). 
+It can be loaded with two lines of code:  
 
-More information needed
+```python
+from diarizers import SegmentationModel
+
+segmentation_model = SegmentationModel().from_pretrained('evie-8/speaker-segmentation-fine-tuned-backup-uganda-eng')
+```
+
+To use it within a pyannote speaker diarization pipeline, load the [pyannote/speaker-diarization-3.1](https://huggingface.co/pyannote/speaker-diarization-3.1) pipeline, and convert the model to a pyannote compatible format: 
+
+```python
+
+from pyannote.audio import Pipeline
+import torch
+
+device = torch.device("cuda:0") if torch.cuda.is_available() else torch.device("cpu")
+
+
+# load the pre-trained pyannote pipeline
+pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization-3.1")
+pipeline.to(device)
+
+# replace the segmentation model with your fine-tuned one
+model = segmentation_model.to_pyannote_model()
+pipeline._segmentation.model = model.to(device)
+```
+
+```python
+# load dataset example
+dataset = load_dataset("evie-8/backup_uganda", "eng", split="data")
+sample = dataset[0]["audio"]
+
+# pre-process inputs
+sample["waveform"] = torch.from_numpy(sample.pop("array")[None, :]).to(device, dtype=model.dtype)
+sample["sample_rate"] = sample.pop("sampling_rate")
+
+# perform inference
+diarization = pipeline(sample)
+
+# dump the diarization output to disk using RTTM format
+with open("audio.rttm", "w") as rttm:
+    diarization.write_rttm(rttm)
+```
 
 ## Intended uses & limitations