Spaces:

KevinGeng
/

Laronix_voice_quality_checking_system_FILEIO

Build error

KevinGeng commited on Oct 9, 2023

Commit

b27c1ab

1 Parent(s): fa64b53

Update app.py

Support multi channels

Files changed (1) hide show

app.py CHANGED Viewed

@@ -44,7 +44,7 @@ class ChangeSampleRate(nn.Module):
 model = lightning_module.BaselineLightningModule.load_from_checkpoint("epoch=3-step=7459.ckpt").eval()
 def calc_mos(audio_path, ref):
-    wav, sr = torchaudio.load(audio_path)
     osr = 16_000
     batch = wav.unsqueeze(0).repeat(10, 1, 1)
     csr = ChangeSampleRate(sr, osr)

 model = lightning_module.BaselineLightningModule.load_from_checkpoint("epoch=3-step=7459.ckpt").eval()
 def calc_mos(audio_path, ref):
+    wav, sr = torchaudio.load(audio_path, channels_first=True)  # Mono channel
     osr = 16_000
     batch = wav.unsqueeze(0).repeat(10, 1, 1)
     csr = ChangeSampleRate(sr, osr)