3loi commited on
Commit
975635a
1 Parent(s): c54d84f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -2
README.md CHANGED
@@ -39,7 +39,8 @@ wavs = torch.tensor(norm_wav).unsqueeze(0)
39
  with torch.no_grad():
40
  pred = model(wavs, mask)
41
 
42
- print(model.config.id2label) #arousal, dominance, valence
43
  print(pred)
44
-
 
45
  ```
 
39
  with torch.no_grad():
40
  pred = model(wavs, mask)
41
 
42
+ print(model.config.id2label)
43
  print(pred)
44
+ #{0: 'arousal', 1: 'dominance', 2: 'valence'}
45
+ #tensor([[0.3670, 0.4553, 0.4240]])
46
  ```