Update README.md
Browse files
README.md
CHANGED
|
@@ -29,7 +29,7 @@ This segmentation model has been trained on English data (Callhome) using diariz
|
|
| 29 |
```python
|
| 30 |
from diarizers import SegmentationModel
|
| 31 |
|
| 32 |
-
segmentation_model = SegmentationModel().from_pretrained('
|
| 33 |
```
|
| 34 |
|
| 35 |
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:
|
|
@@ -47,7 +47,7 @@ device = torch.device("cuda:0") if torch.cuda.is_available() else torch.device("
|
|
| 47 |
pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization-3.1")
|
| 48 |
pipeline.to(device)
|
| 49 |
|
| 50 |
-
model = SegmentationModel().from_pretrained("
|
| 51 |
model = model.to_pyannote_model()
|
| 52 |
pipeline._segmentation.model = model.to(device)
|
| 53 |
```
|
|
@@ -87,7 +87,7 @@ device = torch.device("cuda:0") if torch.cuda.is_available() else torch.device("
|
|
| 87 |
pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization-3.1")
|
| 88 |
pipeline.to(device)
|
| 89 |
|
| 90 |
-
model = SegmentationModel().from_pretrained("
|
| 91 |
model = model.to_pyannote_model()
|
| 92 |
pipeline._segmentation.model = model.to(device)
|
| 93 |
|
|
@@ -131,4 +131,19 @@ The following hyperparameters were used during training:
|
|
| 131 |
- Transformers 4.40.1
|
| 132 |
- Pytorch 2.2.1+cu121
|
| 133 |
- Datasets 2.19.1
|
| 134 |
-
- Tokenizers 0.19.1
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 29 |
```python
|
| 30 |
from diarizers import SegmentationModel
|
| 31 |
|
| 32 |
+
segmentation_model = SegmentationModel().from_pretrained('foduucom/speaker-segmentation-eng')
|
| 33 |
```
|
| 34 |
|
| 35 |
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:
|
|
|
|
| 47 |
pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization-3.1")
|
| 48 |
pipeline.to(device)
|
| 49 |
|
| 50 |
+
model = SegmentationModel().from_pretrained("foduucom/speaker-segmentation-eng")
|
| 51 |
model = model.to_pyannote_model()
|
| 52 |
pipeline._segmentation.model = model.to(device)
|
| 53 |
```
|
|
|
|
| 87 |
pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization-3.1")
|
| 88 |
pipeline.to(device)
|
| 89 |
|
| 90 |
+
model = SegmentationModel().from_pretrained("foduucom/speaker-segmentation-eng")
|
| 91 |
model = model.to_pyannote_model()
|
| 92 |
pipeline._segmentation.model = model.to(device)
|
| 93 |
|
|
|
|
| 131 |
- Transformers 4.40.1
|
| 132 |
- Pytorch 2.2.1+cu121
|
| 133 |
- Datasets 2.19.1
|
| 134 |
+
- Tokenizers 0.19.1
|
| 135 |
+
|
| 136 |
+
|
| 137 |
+
|
| 138 |
+
## Model Card Contact
|
| 139 |
+
|
| 140 |
+
For inquiries and contributions, please contact us at info@foduu.com.
|
| 141 |
+
|
| 142 |
+
```bibtex
|
| 143 |
+
@ModelCard{
|
| 144 |
+
author = {Nehul Agrawal and
|
| 145 |
+
Rahul parihar},
|
| 146 |
+
title = {{Speaker Diarization in english language},
|
| 147 |
+
year = {2023}
|
| 148 |
+
}
|
| 149 |
+
```
|