Spaces:
Sleeping
Sleeping
Remove to(device) call
Browse files- sonogram.py +3 -2
sonogram.py
CHANGED
|
@@ -51,8 +51,9 @@ class Sonogram():
|
|
| 51 |
baselinePipeline.segmentation = ParamDict(min_duration_off=0.0,)
|
| 52 |
self.pipeline = baselinePipeline
|
| 53 |
|
| 54 |
-
|
| 55 |
-
|
|
|
|
| 56 |
|
| 57 |
# Load SVM classifier
|
| 58 |
with open('05062026_groupClassifier.pkl', 'rb') as f:
|
|
|
|
| 51 |
baselinePipeline.segmentation = ParamDict(min_duration_off=0.0,)
|
| 52 |
self.pipeline = baselinePipeline
|
| 53 |
|
| 54 |
+
|
| 55 |
+
# Should manage this outside class now
|
| 56 |
+
#self.pipeline.to(self.device)
|
| 57 |
|
| 58 |
# Load SVM classifier
|
| 59 |
with open('05062026_groupClassifier.pkl', 'rb') as f:
|