Views
No views yet
1from transformers import T5Tokenizer, T5ForConditionalGeneration
2
3tokenizer = T5Tokenizer.from_pretrained("laituan245/molt5-large-caption2smiles", model_max_length=512)
4model = T5ForConditionalGeneration.from_pretrained('laituan245/molt5-large-caption2smiles')
5
6input_text = 'The molecule is a monomethoxybenzene that is 2-methoxyphenol substituted by a hydroxymethyl group at position 4. It has a role as a plant metabolite. It is a member of guaiacols and a member of benzyl alcohols.'
7input_ids = tokenizer(input_text, return_tensors="pt").input_ids
8
9outputs = model.generate(input_ids, num_beams=5, max_length=512)
10print(tokenizer.decode(outputs[0], skip_special_tokens=True))