Komal-patra commited on
Commit
72fd561
·
verified ·
1 Parent(s): 556f86f

update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -34,7 +34,8 @@ 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
 
39
  )
40
  print(outputs)
 
34
  outputs = model.generate(
35
  input_ids = inputs["input_ids"],
36
  max_length=max_length,
37
+ num_beams=7,
38
+ temperature=0.7
39
 
40
  )
41
  print(outputs)