Spaces:
Runtime error
Runtime error
kAIto47802
commited on
Commit
·
a18d920
1
Parent(s):
afa331d
Update app.py
Browse files
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 |
|