Update README.md
This commit is contained in:
parent
2af815790a
commit
e79fe9e308
|
@ -16,7 +16,7 @@ from parrot import Parrot
|
|||
import warnings
|
||||
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?",
|
||||
"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
|
||||
|
||||
para_phrases = parrot.augment(input_phrase=phrase,
|
||||
diversity_ranker="levenshtein"
|
||||
do_diverse=False,
|
||||
max_return_phrases = 10,
|
||||
max_length=32,
|
||||
|
|
Loading…
Reference in New Issue