Spaces:
Runtime error
Runtime error
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]])
|