Update README.md

README.md

@@ -24,7 +24,64 @@ The project of predicting human cognition and emotion, and training details are
 
 You can obtain the personality scores for an input text in the App **[KevSun/Personality_Test]**(https://huggingface.co/spaces/KevSun/Personality_Test).
 
-The following provides the code to implement the task of detecting personality from an input text.
+The following provides the code to implement the task of detecting personality from an input text. However, there are two cases:
+
+The first case doesn't apply softmax and instead outputs the raw logits from the model. 
+It uses **shorter, simpler sentences**. The predicted_scores here are raw logits, which can be any real number and don't sum to 1.
+
+
+```python
+# install these packages before importing them (transformers, PyTorch)
+
+from transformers import AutoModelForSequenceClassification, AutoTokenizer
+import torch
+
+model = AutoModelForSequenceClassification.from_pretrained("KevSun/Personality_LM")
+tokenizer = AutoTokenizer.from_pretrained("KevSun/Personality_LM")
+
+# Choose between direct text input or file input
+use_file = False  # Set to True if you want to read from a file
+
+if use_file:
+    file_path = 'path/to/your/textfile.txt'  # Replace with your file path
+    with open(file_path, 'r', encoding='utf-8') as file:
+        new_text = file.read()
+else:
+    new_text = "I really enjoy working on complex problems and collaborating with others."
+
+# Encode the text using the same tokenizer used during training
+encoded_input = tokenizer(new_text, return_tensors='pt', padding=True, truncation=True, max_length=64)
+
+model.eval()  # Set the model to evaluation mode
+
+# Perform the prediction
+with torch.no_grad():
+    outputs = model(**encoded_input)
+
+
+predictions = outputs.logits.squeeze()
+
+# Convert to numpy array if necessary
+predicted_scores = predictions.numpy()
+
+
+trait_names = ["Agreeableness", "Openness", "Conscientiousness", "Extraversion", "Neuroticism"]
+
+
+for trait, score in zip(trait_names, predicted_scores):
+    print(f"{trait}: {score:.4f}")
+
+##"output":
+#Agreeableness: 0.3965
+#Openness: 0.6714
+#Conscientiousness: 0.3283
+#Extraversion: 0.0026
+#Neuroticism: 0.4645
+
+```
+
+The second case applies softmax to the model outputs, which normalizes the scores into probabilities that sum to 1. It's using longer, more complex sentences and is likely to show more variation in the outputs. 
+The predicted_scores here are probabilities between 0 and 1, and their sum will be 1.
 
 ```python
 # install these packages before importing them (transformers, PyTorch)