Update README.md

README.md

@@ -39,7 +39,8 @@ wavs = torch.tensor(norm_wav).unsqueeze(0)
 with torch.no_grad():
     pred = model(wavs, mask)
 
-print(model.config.id2label) #arousal, dominance, valence
+print(model.config.id2label) 
 print(pred)
-
+#{0: 'arousal', 1: 'dominance', 2: 'valence'}
+#tensor([[0.3670, 0.4553, 0.4240]])
 ```