VladimirVorobev commited on
Commit
c8c5151
1 Parent(s): 5f54179

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -3
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**