Spaces:
Running
on
Zero
Running
on
Zero
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 |
|