Commit
•
1b09024
1
Parent(s):
37abb7c
Update README.md
Browse files
README.md
CHANGED
|
@@ -19,6 +19,7 @@ This model represents the first version of an experimental sequence-to-sequence
|
|
| 19 |
|
| 20 |
## How to Use
|
| 21 |
```python
|
|
|
|
| 22 |
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
| 23 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 24 |
MODEL_NAME = "DeepMount00/OCR_corrector"
|
|
|
|
| 19 |
|
| 20 |
## How to Use
|
| 21 |
```python
|
| 22 |
+
import torch
|
| 23 |
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
| 24 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 25 |
MODEL_NAME = "DeepMount00/OCR_corrector"
|