wetey commited on
Commit
11e365e
·
1 Parent(s): 8983492

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ def generate_headline(text):
13
  input = prefix + text
14
  inputs = tokenizer(input, return_tensors = "pt", max_length = 128, truncation = True).input_ids
15
  generation_config = GenerationConfig(temperature = 1.2,
16
- encoder_no_repeat_ngram_size = 7)
17
  outputs = headline.generate(inputs, do_sample = True, generation_config = generation_config)
18
 
19
  return tokenizer.decode(outputs[0], skip_special_tokens = True)
 
13
  input = prefix + text
14
  inputs = tokenizer(input, return_tensors = "pt", max_length = 128, truncation = True).input_ids
15
  generation_config = GenerationConfig(temperature = 1.2,
16
+ encoder_no_repeat_ngram_size = 4)
17
  outputs = headline.generate(inputs, do_sample = True, generation_config = generation_config)
18
 
19
  return tokenizer.decode(outputs[0], skip_special_tokens = True)