nehulagrawal commited on
Commit
b917115
1 Parent(s): a5f8c33

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +19 -4
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('diarizers-community/speaker-segmentation-fine-tuned-callhome-jpn')
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("nehulagrawal/speaker-segmentation-eng")
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("nehulagrawal/speaker-segmentation-eng")
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
+ ```