kamilakesbi commited on
Commit
60aec22
1 Parent(s): 44733bf

Update README.md

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