Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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=
|
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 |
|