migueldeguzmandev commited on
Commit
b81be42
·
verified ·
1 Parent(s): 481dc55

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ def generate_response(input_text, temperature):
20
  output = model.generate(
21
  input_ids,
22
  attention_mask=attention_mask,
23
- max_length=1024,
24
  num_return_sequences=1,
25
  temperature=temperature,
26
  no_repeat_ngram_size=2,
 
20
  output = model.generate(
21
  input_ids,
22
  attention_mask=attention_mask,
23
+ max_length=300,
24
  num_return_sequences=1,
25
  temperature=temperature,
26
  no_repeat_ngram_size=2,