kAIto47802 commited on
Commit
a18d920
·
1 Parent(s): afa331d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -41,7 +41,7 @@ def predict_mos(audio_path: str, domain: str) -> float:
41
  for _ in range(5):
42
  test_dataset = get_dataset(cfg, data, "test")
43
  p = model(*[torch.tensor(t).unsqueeze(0).to(device) for t in test_dataset[0][:-1]])
44
- preds += p[0]
45
  preds /= 25.0
46
  return preds
47
 
 
41
  for _ in range(5):
42
  test_dataset = get_dataset(cfg, data, "test")
43
  p = model(*[torch.tensor(t).unsqueeze(0).to(device) for t in test_dataset[0][:-1]])
44
+ preds += p.cpu().numpy()[0]
45
  preds /= 25.0
46
  return preds
47