Update README.md

README.md

@@ -33,7 +33,8 @@ The following provides the code to implement the task of detecting personality f
 from transformers import AutoModelForSequenceClassification, AutoTokenizer
 import torch
 
-model = AutoModelForSequenceClassification.from_pretrained("KevSun/Personality_LM")
+warnings.filterwarnings('ignore')
+model = AutoModelForSequenceClassification.from_pretrained("KevSun/Personality_LM", ignore_mismatched_sizes=True)
 tokenizer = AutoTokenizer.from_pretrained("KevSun/Personality_LM")
 
 # Choose between direct text input or file input
@@ -83,8 +84,10 @@ from transformers import AutoModelForSequenceClassification, AutoTokenizer
 import torch
 import argparse
 
+warnings.filterwarnings('ignore')
+
 def load_model_and_tokenizer(model_name):
-    model = AutoModelForSequenceClassification.from_pretrained(model_name)
+    model = AutoModelForSequenceClassification.from_pretrained(model_name, ignore_mismatched_sizes=True)
     tokenizer = AutoTokenizer.from_pretrained(model_name)
     return model, tokenizer