Update README.md

README.md

@@ -29,7 +29,7 @@ tokens = tokenizer("テキスト", truncation=True, return_tensors="pt")
 tokens.to(model.device)
 preds = model(**tokens)
 prob = softmax(preds.logits.cpu().detach().numpy()[0])
-out_dict = {n: p for n, p in zip(emotion_names_jp, prob)}
+out_dict = {n: p for n, p in zip(emotions, prob)}
 
 print(out_dict)
 ```