the-hir0 commited on
Commit
0228405
1 Parent(s): bdf28f7

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -3,8 +3,8 @@ license: apache-2.0
3
  ---
4
  Text Detoxification Task is a process of transforming the text with toxic style into the text with the same meaning but with neutral style
5
 
6
- ```
7
- # Load model directly
8
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
9
 
10
  device = 'cuda' if torch.cuda.is_available() else 'cpu'
 
3
  ---
4
  Text Detoxification Task is a process of transforming the text with toxic style into the text with the same meaning but with neutral style
5
 
6
+ ```python
7
+ import torch
8
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
9
 
10
  device = 'cuda' if torch.cuda.is_available() else 'cpu'