Update README.md

#2
by FrozenWolf - opened
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -6,7 +6,7 @@ tags:
6
  - Conditional Generation
7
  inference: false
8
  ---
9
- ​
10
  # Paraphrase-Generation
11
  ​
12
  ## Model description
@@ -14,14 +14,14 @@ inference: false
14
  T5 Model for generating paraphrases of english sentences. Trained on the [Google PAWS](https://github.com/google-research-datasets/paws) dataset.
15
  ​
16
  ## How to use
17
- ​
18
  PyTorch and TF models available
19
  ​
20
  ```python
21
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
22
  ​
23
  tokenizer = AutoTokenizer.from_pretrained("Vamsi/T5_Paraphrase_Paws")
24
- model = AutoModelForSeq2SeqLM.from_pretrained("Vamsi/T5_Paraphrase_Paws")
25
  ​
26
  sentence = "This is something which i cannot understand at all"
27
 
 
6
  - Conditional Generation
7
  inference: false
8
  ---
9
+
10
  # Paraphrase-Generation
11
  ​
12
  ## Model description
 
14
  T5 Model for generating paraphrases of english sentences. Trained on the [Google PAWS](https://github.com/google-research-datasets/paws) dataset.
15
  ​
16
  ## How to use
17
+ ​## Requires sentencepiece: # !pip install sentencepiece
18
  PyTorch and TF models available
19
  ​
20
  ```python
21
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
22
  ​
23
  tokenizer = AutoTokenizer.from_pretrained("Vamsi/T5_Paraphrase_Paws")
24
+ model = AutoModelForSeq2SeqLM.from_pretrained("Vamsi/T5_Paraphrase_Paws").to('cuda')
25
  ​
26
  sentence = "This is something which i cannot understand at all"
27