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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -34,8 +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=5,
38
- num_return_sequences=3
39
  )
40
 
41
  print(outputs)
 
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)