Update README.md

README.md

@@ -9,13 +9,11 @@ from transformers import AutoTokenizer, AutoModel
 
 model_name = "brjoey/climatebert_econbert" 
 tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModel.from_pretrained(model_name)
 
-# Load the model directly with
-from transformers import AutoModel
-model = AutoModel.from_pretrained("brjoey/climatebert_econbert", torch_dtype="auto")
+# Load the base model
+model = AutoModel.from_pretrained(model_name, torch_dtype="auto")
 
-# For sequence classification
+# For sequence classification tasks
 from transformers import AutoModelForSequenceClassification
 model = AutoModelForSequenceClassification.from_pretrained(model_name, num_labels=3)
 ```