Replace str with torch.device

sonogram.py

@@ -8,6 +8,7 @@ from pyannote.core import Segment
 from pyannote.audio.models.segmentation import PyanNet
 from pyannote.audio import Inference
 from pyannote.pipeline.parameter import ParamDict
+import torch
 
 class Sonogram():
 
@@ -226,5 +227,5 @@ class Sonogram():
         print(f"Sonogram moved to {self.device}")
 
     def toCPU(self):
-        self.pipeline.to('cpu')
+        self.pipeline.to(torch.device('cpu'))
         print(f"Sonogram moved to CPU")