prithivida
/
parrot_paraphraser_on_T5
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update README.md

This commit is contained in:
Prithiviraj Damodaran 2021-05-07 13:55:20 +00:00 committed by huggingface-web
parent d66575dc04
commit ffa21e765f
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
README.md
View File

@ -12,18 +12,22 @@ pip install git+https://github.com/PrithivirajDamodaran/Parrot.git
### Quickstart ### Quickstart
```python ```python
# Caveat: the generate part is NOT seeded, so for a same input, multiple runs will produce DIFFERENT outputs for now
from parrot import Parrot from parrot import Parrot
import warnings import warnings
warnings.filterwarnings("ignore") warnings.filterwarnings("ignore")
#Init models (make sure you init ONLY once if you integrate this to your code)
parrot = Parrot(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?"
] ]
for phrase in phrases: for phrase in phrases:
print("-"*100) print("-"*100)
print(phrase) print("Input_phrase: ", phrase)
print("-"*100) print("-"*100)
para_phrases = parrot.augment(input_phrase=phrase) para_phrases = parrot.augment(input_phrase=phrase)
for para_phrase in para_phrases: for para_phrase in para_phrases:
@ -31,8 +35,6 @@ for phrase in phrases:
``` ```
<pre> <pre>
----------------------------------------------------------------------------- -----------------------------------------------------------------------------
Input_phrase: Can you recommed some upscale restaurants in Rome? Input_phrase: Can you recommed some upscale restaurants in Rome?
----------------------------------------------------------------------------- -----------------------------------------------------------------------------
@ -54,7 +56,6 @@ Input_phrase: What are the famous places we should not miss in Russia
"what are some of the most important places to visit in russia?" "what are some of the most important places to visit in russia?"
"what are some of the most famous places of russia?" "what are some of the most famous places of russia?"
"what are some places we should not miss in russia?" "what are some places we should not miss in russia?"
</pre> </pre>
### Knobs ### Knobs