Update app.py
Browse files
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))
|