Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -8,7 +8,7 @@ def codegen(intent):
|
|
8 |
text = " write a function which takes 2 numbers as input and returns the larger of the two"
|
9 |
input_ids = codegen_tkn(intent, return_tensors="pt").input_ids
|
10 |
|
11 |
-
gen_ids = mdl.generate(input_ids, max_length=
|
12 |
response = codegen_tkn.decode(gen_ids[0], skip_special_tokens=True)
|
13 |
return response
|
14 |
|
|
|
8 |
text = " write a function which takes 2 numbers as input and returns the larger of the two"
|
9 |
input_ids = codegen_tkn(intent, return_tensors="pt").input_ids
|
10 |
|
11 |
+
gen_ids = mdl.generate(input_ids, max_length=512)
|
12 |
response = codegen_tkn.decode(gen_ids[0], skip_special_tokens=True)
|
13 |
return response
|
14 |
|