xferdie commited on
Commit
7695190
·
verified ·
1 Parent(s): e4c9e6f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -23,8 +23,6 @@ text = """ public static void main(String[] args) {
23
  } """
24
 
25
  input_ids = tokenizer(text, return_tensors="pt").input_ids
26
-
27
- input_ids = codeT5_tkn(text, return_tensors="pt").input_ids
28
- generated_ids = mdl.generate(input_ids, max_length=20)
29
 
30
  print(tokenizer.decode(generated_ids[0], skip_special_tokens=True))
 
23
  } """
24
 
25
  input_ids = tokenizer(text, return_tensors="pt").input_ids
26
+ generated_ids = model.generate(input_ids, max_length=20)
 
 
27
 
28
  print(tokenizer.decode(generated_ids[0], skip_special_tokens=True))