Commit
•
c8c5151
1
Parent(s):
5f54179
Update README.md
Browse files
README.md
CHANGED
@@ -18,10 +18,10 @@ This model is based on the T5-base model. We used "transfer learning" to get our
|
|
18 |
|
19 |
**Deploying example:**
|
20 |
```python
|
21 |
-
device = "cuda"
|
22 |
-
|
23 |
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
24 |
|
|
|
|
|
25 |
tokenizer = AutoTokenizer.from_pretrained("humarin/chatgpt_paraphraser_on_T5_base")
|
26 |
|
27 |
model = AutoModelForSeq2SeqLM.from_pretrained("humarin/chatgpt_paraphraser_on_T5_base").to(device)
|
@@ -42,7 +42,6 @@ def paraphrase(text, max_length=128, num_return_sequences=5, num_beams=25, tempe
|
|
42 |
res = tokenizer.batch_decode(outputs, skip_special_tokens=True)
|
43 |
|
44 |
return res
|
45 |
-
|
46 |
```
|
47 |
|
48 |
**Usage examples**
|
|
|
18 |
|
19 |
**Deploying example:**
|
20 |
```python
|
|
|
|
|
21 |
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
22 |
|
23 |
+
device = "cuda"
|
24 |
+
|
25 |
tokenizer = AutoTokenizer.from_pretrained("humarin/chatgpt_paraphraser_on_T5_base")
|
26 |
|
27 |
model = AutoModelForSeq2SeqLM.from_pretrained("humarin/chatgpt_paraphraser_on_T5_base").to(device)
|
|
|
42 |
res = tokenizer.batch_decode(outputs, skip_special_tokens=True)
|
43 |
|
44 |
return res
|
|
|
45 |
```
|
46 |
|
47 |
**Usage examples**
|