Update README.md

README.md

@@ -534,7 +534,7 @@ from transformers import AutoTokenizer, AutoModel, utils
 from bertviz import model_view
 utils.logging.set_verbosity_error()  # Suppress standard warnings
 
-model_name = "./mlm/2-chunk-3o"  
+model_name = "gbyuvd/chemselfies-base-bertmlm"  
 input_text = "[C] [C] [=Branch1] [C] [=O] [O] [C] [C] [N+1] [Branch1] [C] [C] [Branch1] [C] [C] [C]"
 model = AutoModel.from_pretrained(model_name, output_attentions=True)  # Configure model to return attention values
 tokenizer = AutoTokenizer.from_pretrained(model_name)