Jaehan commited on
Commit
8c5dd78
1 Parent(s): d3c2d2f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ def codegen(intent):
11
  #text = "Write a function which takes 2 numbers as input and returns the larger of the two."
12
 
13
  input_ids = codegen_token(intent, return_tensors="pt").input_ids
14
- outcode_ids = model.generate(input_ids, max_length=128)
15
  response = codegen_token.decode(outcode_ids[0], skip_special_tokens=True)
16
  return response
17
 
 
11
  #text = "Write a function which takes 2 numbers as input and returns the larger of the two."
12
 
13
  input_ids = codegen_token(intent, return_tensors="pt").input_ids
14
+ outcode_ids = model.generate(input_ids, max_length=256)
15
  response = codegen_token.decode(outcode_ids[0], skip_special_tokens=True)
16
  return response
17