suriya7 commited on
Commit
ae744dd
1 Parent(s): 3e11aff

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -46,7 +46,7 @@ def run_generation(user_text, top_p, temperature, top_k, max_new_tokens):
46
  do_sample=True,
47
  repetition_penalty=1.5,
48
  temperature=0.7,
49
- top_k=0.2,
50
  )
51
  t = Thread(target=model.generate, kwargs=generate_kwargs)
52
  t.start()
 
46
  do_sample=True,
47
  repetition_penalty=1.5,
48
  temperature=0.7,
49
+ top_k=2,
50
  )
51
  t = Thread(target=model.generate, kwargs=generate_kwargs)
52
  t.start()