xferdie commited on
Commit
8927c0c
·
verified ·
1 Parent(s): ca29e6b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,6 +1,6 @@
1
- from transformers import RobertaTokenizer, T5ForConditionalGeneration
2
 
3
- tokenizer = RobertaTokenizer.from_pretrained('Salesforce/codet5-small')
4
  model = T5ForConditionalGeneration.from_pretrained('Salesforce/codet5-small')
5
 
6
  text = "def greet(user): print(f'hello <extra_id_0>!')"
 
1
+ from transformers import T5Tokenizer, T5ForConditionalGeneration
2
 
3
+ tokenizer = T5Tokenizer.from_pretrained('Salesforce/codet5-small')
4
  model = T5ForConditionalGeneration.from_pretrained('Salesforce/codet5-small')
5
 
6
  text = "def greet(user): print(f'hello <extra_id_0>!')"