upperwal commited on
Commit
9546834
·
verified ·
1 Parent(s): 0e9349e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -38,7 +38,7 @@ def predict(message, history):
38
  generate_kwargs = dict(
39
  model_inputs,
40
  streamer=streamer,
41
- max_new_tokens=1024,
42
  do_sample=True,
43
  top_p=0.95,
44
  top_k=50,
 
38
  generate_kwargs = dict(
39
  model_inputs,
40
  streamer=streamer,
41
+ max_new_tokens=50,
42
  do_sample=True,
43
  top_p=0.95,
44
  top_k=50,