Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
@@ -39,7 +39,7 @@ def single_predict(audio_path, domain, quick):
|
|
39 |
spec = compute_spec(wav)
|
40 |
preds = []
|
41 |
# kui sul on fine-tuningus kasutatud nt 8 domääni, muuda siit
|
42 |
-
NUM_DOMAINS =
|
43 |
for dom in range(NUM_DOMAINS):
|
44 |
d_oh = torch.nn.functional.one_hot(torch.tensor(dom, device=DEVICE),
|
45 |
num_classes=model.num_dataset).float()[None]
|
|
|
39 |
spec = compute_spec(wav)
|
40 |
preds = []
|
41 |
# kui sul on fine-tuningus kasutatud nt 8 domääni, muuda siit
|
42 |
+
NUM_DOMAINS = 3
|
43 |
for dom in range(NUM_DOMAINS):
|
44 |
d_oh = torch.nn.functional.one_hot(torch.tensor(dom, device=DEVICE),
|
45 |
num_classes=model.num_dataset).float()[None]
|