pszemraj commited on
Commit
0240ed4
1 Parent(s): 9a08475

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -29,6 +29,7 @@ def run_generation(user_text, top_p, temperature, top_k, max_new_tokens, use_gen
29
  streamer=streamer,
30
  max_new_tokens=max_new_tokens,
31
  do_sample=True,
 
32
  top_p=top_p,
33
  temperature=float(temperature),
34
  top_k=top_k
 
29
  streamer=streamer,
30
  max_new_tokens=max_new_tokens,
31
  do_sample=True,
32
+ num_beams=1,
33
  top_p=top_p,
34
  temperature=float(temperature),
35
  top_k=top_k