vilarin commited on
Commit
1fc0e10
·
verified ·
1 Parent(s): a983a99

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -68,6 +68,8 @@ def translate(
68
  max_length=max_length,
69
  do_sample=True,
70
  temperature=temperature,
 
 
71
  )
72
 
73
  outputs = model.generate(**generate_kwargs)
 
68
  max_length=max_length,
69
  do_sample=True,
70
  temperature=temperature,
71
+ top_p=top_p,
72
+ repetition_penalty=rp,
73
  )
74
 
75
  outputs = model.generate(**generate_kwargs)