Update paraphraser.py
Browse files- 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=
|
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",
|