Update README.md

README.md

@@ -54,10 +54,10 @@ model.eval()  # Set the model to evaluation mode
 with torch.no_grad():
     outputs = model(**encoded_input)
 
-predictions = outputs.logits.squeeze()
 
-# Convert to numpy array if necessary
-predicted_scores = predictions.numpy()
+predictions = torch.nn.functional.softmax(outputs.logits, dim=-1)
+predicted_scores = predictions[0].tolist()
+
 
 trait_names = ["Agreeableness", "Openness", "Conscientiousness", "Extraversion", "Neuroticism"]