Komal-patra commited on
Commit
473cfba
·
verified ·
1 Parent(s): bd66ea9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -34,8 +34,9 @@ def generate_text(prompt, max_length=200):
34
  outputs = model.generate(
35
  input_ids = inputs["input_ids"],
36
  max_length=max_length,
37
- num_beams=1,
38
- num_return_sequences=1
 
39
  )
40
 
41
  print(outputs)
 
34
  outputs = model.generate(
35
  input_ids = inputs["input_ids"],
36
  max_length=max_length,
37
+ num_beams=3,
38
+ num_return_sequences=1,
39
+ temperature=0.9
40
  )
41
 
42
  print(outputs)