monatolmats commited on
Commit
dbf8a9e
·
verified ·
1 Parent(s): dc1aa8f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = 8
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]