shuttie commited on
Commit
604080b
1 Parent(s): 957486f

max tokens 128

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ prompt = "### Instruction:\nContinue a dad joke:\n\n### Input:\n{input}\n\n### R
17
 
18
  def make_response(message, history):
19
  input = prompt.format(input=message)
20
- generated = generator(input, return_full_text=False, max_new_tokens=64, num_return_sequences=1)
21
  return generated[0]["generated_text"]
22
 
23
 
 
17
 
18
  def make_response(message, history):
19
  input = prompt.format(input=message)
20
+ generated = generator(input, return_full_text=False, max_new_tokens=128, num_return_sequences=1)
21
  return generated[0]["generated_text"]
22
 
23