Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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]])
|