Update README.md
Browse files
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)
|
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 |
```
|