Update README.md

README.md

@@ -6,7 +6,7 @@ tags:
 - Conditional Generation
 inference: false
 ---
-​
+
 # Paraphrase-Generation
 ​
 ## Model description
@@ -14,14 +14,14 @@ inference: false
 T5 Model for generating paraphrases of english sentences. Trained on the [Google PAWS](https://github.com/google-research-datasets/paws) dataset.
 ​
 ## How to use
-​
+​## Requires sentencepiece: # !pip install sentencepiece
 PyTorch and TF models available
 ​
 ```python
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 ​
 tokenizer = AutoTokenizer.from_pretrained("Vamsi/T5_Paraphrase_Paws")  
-model = AutoModelForSeq2SeqLM.from_pretrained("Vamsi/T5_Paraphrase_Paws")
+model = AutoModelForSeq2SeqLM.from_pretrained("Vamsi/T5_Paraphrase_Paws").to('cuda')
 ​
 sentence = "This is something which i cannot understand at all"