Wataru commited on
Commit
afa331d
·
verified ·
1 Parent(s): bba8561

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -37,7 +37,7 @@ def predict_mos(audio_path: str, domain: str) -> float:
37
  preds = 0.0
38
  for fold in range(5):
39
  cfg.now_fold = fold
40
- model = get_model(cfg, device)
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]])
 
37
  preds = 0.0
38
  for fold in range(5):
39
  cfg.now_fold = fold
40
+ model = get_model(cfg, device).eval()
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]])