nehulagrawal
commited on
Commit
•
b917115
1
Parent(s):
a5f8c33
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 [email protected].
|
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 |
+
```
|