run480 commited on
Commit
bf125a7
·
verified ·
1 Parent(s): 7a1062c

Update app.py

Browse files

Increase generated text length and avoid repetition in the generated output.

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -171,7 +171,7 @@ gpt2_tkn=GPT2Tokenizer.from_pretrained('gpt2')
171
 
172
  def generate(starting_text):
173
  tkn_ids = gpt2_tkn.encode(starting_text, return_tensors = 'pt')
174
- gpt2_tensors = mdl.generate(tkn_ids)
175
  response=""
176
  #response = gpt2_tensors
177
  for i, x in enumerate(gpt2_tensors):
 
171
 
172
  def generate(starting_text):
173
  tkn_ids = gpt2_tkn.encode(starting_text, return_tensors = 'pt')
174
+ gpt2_tensors = mdl.generate(tkn_ids, max_length=100, no_repeat_ngram_size=True)
175
  response=""
176
  #response = gpt2_tensors
177
  for i, x in enumerate(gpt2_tensors):