prithivMLmods commited on
Commit
f9c1d72
·
verified ·
1 Parent(s): b2f9493

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -80,7 +80,7 @@ def model_inference(input_dict, history):
80
  generation_kwargs = dict(
81
  inputs,
82
  streamer=streamer,
83
- max_new_tokens=512,
84
  do_sample=True,
85
  temperature=0.3
86
  )
 
80
  generation_kwargs = dict(
81
  inputs,
82
  streamer=streamer,
83
+ max_new_tokens=1024,
84
  do_sample=True,
85
  temperature=0.3
86
  )