Update README.md
Browse files
README.md
CHANGED
@@ -53,7 +53,7 @@ inputs = tokenizer("This is a test input.", return_tensors="pt")
|
|
53 |
outputs = model(inputs['input_ids'], inputs['attention_mask'])
|
54 |
|
55 |
# Print out the emotion ratings
|
56 |
-
for emotion, rating in zip(['Valence', 'Arousal', 'Dominance', 'Origin', 'Significance', 'Concreteness', 'Imageability', 'Acquisition'], outputs):
|
57 |
print(f"{emotion}: {rating.item()}")
|
58 |
```
|
59 |
|
|
|
53 |
outputs = model(inputs['input_ids'], inputs['attention_mask'])
|
54 |
|
55 |
# Print out the emotion ratings
|
56 |
+
for emotion, rating in zip(['Valence', 'Arousal', 'Dominance', 'Origin', 'Significance', 'Concreteness', 'Imageability', 'Age of Acquisition'], outputs):
|
57 |
print(f"{emotion}: {rating.item()}")
|
58 |
```
|
59 |
|