Replace YAMNet model for panns-inference

YAMNet model was causing issues, as loading a pytorch framework model, when the tensorflow based YAMNet is loaded in the same environment already, caused segmentation fault
2025-02-26 17:36:27 +05:30
parent f2011b4408
commit cbf2b022a5
4 changed files with 70 additions and 160 deletions
@@ -24,7 +24,7 @@ def test_trim_audio():

 def test_classify():
    file_path = "tests/test_audio/cafe_crowd_talk.wav"
-    waveform, sr = read_audio(file_path, 16000, mono=True)
+    waveform, sr = read_audio(file_path, 32000, mono=True)
    expected_class = "Speech"
    predicted_class = classify_audio(waveform, sr)