From e79fe9e3088ab2850ecf3dc81718e527f0e5efcc Mon Sep 17 00:00:00 2001 From: Prithiviraj Damodaran Date: Fri, 7 May 2021 13:12:28 +0000 Subject: [PATCH] Update README.md --- README.md | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index b899319..f575e9e 100644 --- a/README.md +++ b/README.md @@ -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,