jgyasu commited on
Commit
c98e974
·
verified ·
1 Parent(s): 960f419

Update paraphraser.py

Browse files
Files changed (1) hide show
  1. paraphraser.py +1 -1
paraphraser.py CHANGED
@@ -7,7 +7,7 @@ def init_model():
7
  return para_tokenizer, para_model
8
 
9
  # Function to Paraphrase the Text
10
- def paraphrase(question, para_tokenizer, para_model, num_beams=5, num_beam_groups=5, num_return_sequences=5, repetition_penalty=10.0, diversity_penalty=3.0, no_repeat_ngram_size=2, temperature=0.7, max_length=64):
11
  input_ids = para_tokenizer(
12
  f'paraphrase: {question}',
13
  return_tensors="pt", padding="longest",
 
7
  return para_tokenizer, para_model
8
 
9
  # Function to Paraphrase the Text
10
+ def paraphrase(question, para_tokenizer, para_model, num_beams=10, num_beam_groups=10, num_return_sequences=10, repetition_penalty=10.0, diversity_penalty=3.0, no_repeat_ngram_size=2, temperature=0.7, max_length=64):
11
  input_ids = para_tokenizer(
12
  f'paraphrase: {question}',
13
  return_tensors="pt", padding="longest",