Update README.md

README.md

@@ -30,7 +30,6 @@ The following provides the code to implement the task of detecting personality f
 ```python
 # install these packages before importing them (transformers, PyTorch)
 # install these packages before importing them (transformers, PyTorch)
-
 from transformers import AutoModelForSequenceClassification, AutoTokenizer
 import torch
 
@@ -61,18 +60,19 @@ predictions = torch.nn.functional.softmax(outputs.logits, dim=-1)
 predicted_scores = predictions[0].tolist()
 
 
-trait_names = ["Agreeableness", "Openness", "Conscientiousness", "Extraversion", "Neuroticism"]
+trait_names = ["agreeableness", "openness", "conscientiousness", "extraversion", "neuroticism"]
 
 
 for trait, score in zip(trait_names, predicted_scores):
     print(f"{trait}: {score:.4f}")
 
 ##"output":
-#Agreeableness: 0.3032
-#Openness: 0.1607
-#Conscientiousness: 0.1835
-#Extraversion: 0.1558
-#Neuroticism: 0.1969
+#agreeableness: 0.2138
+#openness: 0.2890
+#conscientiousness: 0.1921
+#extraversion: 0.1307
+#neuroticism: 0.1744
+
 
 
 ```