Update README.md

README.md

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