hplisiecki commited on
Commit
da18575
·
verified ·
1 Parent(s): 80d247b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -52,7 +52,7 @@ inputs = tokenizer("This is a test input.", return_tensors="pt")
52
  outputs = model(inputs['input_ids'], inputs['attention_mask'])
53
 
54
  # Print out the emotion ratings
55
- for emotion, rating in zip(['valence', 'arousal', 'dominance', 'aoa', 'concreteness'], outputs):
56
  print(f"{emotion}: {rating.item()}")
57
  ```
58
 
 
52
  outputs = model(inputs['input_ids'], inputs['attention_mask'])
53
 
54
  # Print out the emotion ratings
55
+ for emotion, rating in zip(['Valence', 'Arousal', 'Dominance', 'Age of Acquisition', 'Concreteness'], outputs):
56
  print(f"{emotion}: {rating.item()}")
57
  ```
58