boryasbora commited on
Commit
42449f3
·
verified ·
1 Parent(s): 681e723

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -81,7 +81,8 @@ def get_chain(temperature):
81
  "text-generation",
82
  model=model,
83
  tokenizer=tokenizer,
84
- max_length=2048,
 
85
  temperature=temperature,
86
  top_p=0.95,
87
  repetition_penalty=1.15
 
81
  "text-generation",
82
  model=model,
83
  tokenizer=tokenizer,
84
+ max_length=4000,
85
+ max_new_tokens = 500,
86
  temperature=temperature,
87
  top_p=0.95,
88
  repetition_penalty=1.15