kamilakesbi commited on
Commit
0c244ec
1 Parent(s): 414dca9

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -51,7 +51,7 @@ pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization-3.1")
51
  pipeline.to(device)
52
 
53
  # replace the segmentation model with your fine-tuned one
54
- segmentation_model = segmentation_model.to_pyannote_model()
55
  pipeline._segmentation.model = model.to(device)
56
  ```
57
 
 
51
  pipeline.to(device)
52
 
53
  # replace the segmentation model with your fine-tuned one
54
+ model = segmentation_model.to_pyannote_model()
55
  pipeline._segmentation.model = model.to(device)
56
  ```
57