Update README.md

This commit is contained in:
Prithiviraj Damodaran 2021-05-07 13:12:28 +00:00 committed by huggingface-web
parent 2af815790a
commit e79fe9e308
1 changed files with 2 additions and 1 deletions

View File

@ -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,