LejobuildYT/codeT5-DC

1from transformers import RobertaTokenizer, T5ForConditionalGeneration2 3tokenizer = RobertaTokenizer.from_pretrained('Salesforce/codet5-small')4model = T5ForConditionalGeneration.from_pretrained('Salesforce/codet5-small')5 6#def greet(user): print(f'hello <extra_id_0>!')"7text = input("")8input_ids = tokenizer(text, return_tensors="pt").input_ids9 10# simply generate a single sequence11generated_ids = model.generate(input_ids, max_length=10)12print(tokenizer.decode(generated_ids[0], skip_special_tokens=True))13# this prints "user: {user.name}"14