Update README.md

phosseini · Apr 29, 2022 · 841bd81 · 841bd81
1 parent 4953eaf
commit 841bd81
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -30,7 +30,9 @@ from transformers import AutoTokenizer, AutoModel
 
 tokenizer = AutoTokenizer.from_pretrained("bert-large-cased")
 
-atomic_model = AutoModel.from_pretrained("phosseini/atomic-bert-large")
+atomic_bert_model = AutoModel.from_pretrained("phosseini/atomic-bert-large")
+atomic_roberta_model = AutoModel.from_pretrained("phosseini/atomic-roberta-large")
+
 glucose_model = AutoModel.from_pretrained("phosseini/glucose-bert-large")
 ```