Spaces:

Sonogram
/

Instructor-Support-Tool

Running on CPU Upgrade

duongthienz commited on 12 days ago

Commit

9d04df1

verified ·

1 Parent(s): 2bc81ff

Update sonogram.py

Files changed (1) hide show

sonogram.py CHANGED Viewed

@@ -84,7 +84,7 @@ class Sonogram():
         elif version == '1.0':
             baselinePipeline = Pipeline.from_pretrained("pyannote/speaker-diarization-3.1")
             newSpecs = baselinePipeline._segmentation.model.specifications
-            segModel = PyanNet.from_pretrained('20251208_Sonogram_Segmentation.ckpt')
             segModel.specifications = newSpecs
             segmentation_duration = segModel.specifications.duration

         elif version == '1.0':
             baselinePipeline = Pipeline.from_pretrained("pyannote/speaker-diarization-3.1")
             newSpecs = baselinePipeline._segmentation.model.specifications
+            segModel = PyanNet.from_pretrained('20260810_Sonogram_Segmentation_v2.ckpt')
             segModel.specifications = newSpecs
             segmentation_duration = segModel.specifications.duration