Commit ·
e5d293a
1
Parent(s): 51a9e2d
Update README.md
Browse files
README.md
CHANGED
|
@@ -51,8 +51,12 @@ pip install bert-for-sequence classfication
|
|
| 51 |
|
| 52 |
```python
|
| 53 |
from bert_clf import EncoderCLF
|
|
|
|
| 54 |
|
| 55 |
model = EncoderCLF("whitefoxredhell/language_identification")
|
|
|
|
|
|
|
|
|
|
| 56 |
|
| 57 |
text = "London is the capital of Great Britain"
|
| 58 |
|
|
|
|
| 51 |
|
| 52 |
```python
|
| 53 |
from bert_clf import EncoderCLF
|
| 54 |
+
import torch
|
| 55 |
|
| 56 |
model = EncoderCLF("whitefoxredhell/language_identification")
|
| 57 |
+
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 58 |
+
model = model.to(device)
|
| 59 |
+
model = model.eval()
|
| 60 |
|
| 61 |
text = "London is the capital of Great Britain"
|
| 62 |
|