Update app.py

app.py

@@ -39,7 +39,7 @@ def predict_mos(audio_path: str, domain: str) -> float:
         model = get_model(cfg, device)
         for _ in range(5):
             test_dataset = get_dataset(cfg, data, "test")
-            p = model(*[torch.tensor(t).unsqueeze(0) for t in test_dataset[0][:-1]])
+            p = model(*[torch.tensor(t).unsqueeze(0).to(device) for t in test_dataset[0][:-1]])
             preds += p[0]
     preds /= 25.0
     return preds