Update README.md

README.md

@@ -17,7 +17,7 @@ molxpt_tokenizer = AutoTokenizer.from_pretrained("zequnl/molxpt", trust_remote_c
 model = model.cuda()
 model.eval()
 
-input_ids = molxpt_tokenizer('Aspirin is', return_tensors="pt").input_ids.cuda()
+input_ids = molxpt_tokenizer('<start-of-mol>CC(=O)OC1=CC=CC=C1C(=O)O<end-of-mol> is ', return_tensors="pt").input_ids.cuda()
 output = model.generate(
     input_ids,
     max_new_tokens=300,