Spaces:
Sleeping
Sleeping
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
|