Komal-patra commited on
Commit
64ba630
1 Parent(s): 2170956
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -35,7 +35,7 @@ def generate_text(prompt, max_length=512):
35
  input_ids = inputs["input_ids"],
36
  max_length=max_length,
37
  num_beams=1,
38
- temperature=0.1
39
  )
40
 
41
  print(outputs)
 
35
  input_ids = inputs["input_ids"],
36
  max_length=max_length,
37
  num_beams=1,
38
+ repetition_penalty=1.0
39
  )
40
 
41
  print(outputs)