LejobuildYT commited on
Commit
530bcd8
·
verified ·
1 Parent(s): 0853ad3

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -0
app.py ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ #def greet(user): print(f'hello <extra_id_0>!')"
7
+ text = input("")
8
+ input_ids = tokenizer(text, return_tensors="pt").input_ids
9
+
10
+ # simply generate a single sequence
11
+ generated_ids = model.generate(input_ids, max_length=10)
12
+ print(tokenizer.decode(generated_ids[0], skip_special_tokens=True))
13
+ # this prints "user: {user.name}"