Spaces:
Build error
Build error
Commit ·
2dd2a84
1
Parent(s): 696d1c3
Update app.py
Browse files
app.py
CHANGED
|
@@ -9,7 +9,7 @@ def codegen(intent):
|
|
| 9 |
# text = " write a function which takes 2 numbers as input
|
| 10 |
# and returns the larger of the two"
|
| 11 |
input_ids = codegen_tkn(intent, return_tensors = 'pt').input_ids
|
| 12 |
-
gen_ids = mdl.generate(input_ids, max_length =
|
| 13 |
response = codegen_tkn.decode(gen_ids[0], skip_special_tokens = True)
|
| 14 |
|
| 15 |
return response
|
|
|
|
| 9 |
# text = " write a function which takes 2 numbers as input
|
| 10 |
# and returns the larger of the two"
|
| 11 |
input_ids = codegen_tkn(intent, return_tensors = 'pt').input_ids
|
| 12 |
+
gen_ids = mdl.generate(input_ids, max_length = 1024)
|
| 13 |
response = codegen_tkn.decode(gen_ids[0], skip_special_tokens = True)
|
| 14 |
|
| 15 |
return response
|