hplisiecki
commited on
Update README.md
Browse files
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(['
|
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 |
|