Update README.md
This commit is contained in:
parent
2af815790a
commit
e79fe9e308
|
@ -16,7 +16,7 @@ from parrot import Parrot
|
||||||
import warnings
|
import warnings
|
||||||
warnings.filterwarnings("ignore")
|
warnings.filterwarnings("ignore")
|
||||||
|
|
||||||
parrot = Parrot(diversity_ranker="levenshtein", model_tag="prithivida/parrot_paraphraser_on_T5", use_gpu=False)
|
parrot = Parrot(model_tag="prithivida/parrot_paraphraser_on_T5", use_gpu=False)
|
||||||
phrases = ["Can you recommed some upscale restaurants in Rome?",
|
phrases = ["Can you recommed some upscale restaurants in Rome?",
|
||||||
"What are the famous places we should not miss in Russia?"
|
"What are the famous places we should not miss in Russia?"
|
||||||
]
|
]
|
||||||
|
@ -62,6 +62,7 @@ Input_phrase: What are the famous places we should not miss in Russia
|
||||||
```python
|
```python
|
||||||
|
|
||||||
para_phrases = parrot.augment(input_phrase=phrase,
|
para_phrases = parrot.augment(input_phrase=phrase,
|
||||||
|
diversity_ranker="levenshtein"
|
||||||
do_diverse=False,
|
do_diverse=False,
|
||||||
max_return_phrases = 10,
|
max_return_phrases = 10,
|
||||||
max_length=32,
|
max_length=32,
|
||||||
|
|
Loading…
Reference in New Issue