Update README.md

README.md

@@ -24,8 +24,11 @@ ROC-AUC score: 0.86
 ## Usage:
 ```python
 from transformers import T5ForConditionalGeneration, T5Tokenizer
+import torch
+# use GPU for better performance
+device = torch.device('cuda')
 tokenizer = T5Tokenizer.from_pretrained("etomoscow/T5_paraphrase_detector")
-model = T5ForConditionalGeneration.from_pretrained("etomoscow/T5_paraphrase_detector")
+model = T5ForConditionalGeneration.from_pretrained("etomoscow/T5_paraphrase_detector").to(device)
 text_1 = 'During her sophomore , junior and senior summers , she spent half of it with her Alaska team , and half playing , and living in Oregon .'
 text_2 = 'During her second , junior and senior summers , she spent half of it with her Alaska team , half playing and living in Oregon.'
 true_label = '1'