kamilakesbi commited on
Commit
702d3d1
1 Parent(s): 84f3b49

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -51,7 +51,7 @@ pipeline.to(device)
51
 
52
  # replace the segmentation model with your fine-tuned one
53
  segmentation_model = segmentation_model.to_pyannote_model()
54
- pipeline._segmentation.model = model.to(device)
55
  ```
56
 
57
  You can now use the pipeline on audio examples:
 
51
 
52
  # replace the segmentation model with your fine-tuned one
53
  segmentation_model = segmentation_model.to_pyannote_model()
54
+ pipeline._segmentation.model = segmentation_model.to(device)
55
  ```
56
 
57
  You can now use the pipeline on audio examples: