NoaiGPT commited on
Commit
4bc8ae4
1 Parent(s): 5096125
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -274,13 +274,13 @@ def generate_paraphrases(text, setting, output_format):
274
  for i, sentence in enumerate(sentences):
275
  paraphrases = paraphrase_pipeline(
276
  sentence,
277
- num_beams=5,
278
- num_beam_groups=1,
279
  num_return_sequences=num_return_sequences,
280
- diversity_penalty=1.0,
 
 
 
281
  no_repeat_ngram_size=no_repeat_ngram_size,
282
  repetition_penalty=repetition_penalty,
283
- temperature=temperature,
284
  max_length=max_length
285
  )
286
 
 
274
  for i, sentence in enumerate(sentences):
275
  paraphrases = paraphrase_pipeline(
276
  sentence,
 
 
277
  num_return_sequences=num_return_sequences,
278
+ do_sample=True,
279
+ top_k=top_k,
280
+ top_p=top_p,
281
+ temperature=temperature,
282
  no_repeat_ngram_size=no_repeat_ngram_size,
283
  repetition_penalty=repetition_penalty,
 
284
  max_length=max_length
285
  )
286