Commit ·
df8f4a0
1
Parent(s): 0cdfe4d
Update README.md
Browse files
README.md
CHANGED
|
@@ -28,7 +28,6 @@ model = AutoModelForSequenceClassification.from_pretrained("serdarakyol/interpre
|
|
| 28 |
|
| 29 |
```sh
|
| 30 |
import torch
|
| 31 |
-
import numpy as np
|
| 32 |
|
| 33 |
if torch.cuda.is_available():
|
| 34 |
device = torch.device("cuda")
|
|
@@ -40,6 +39,8 @@ else:
|
|
| 40 |
device = torch.device("cpu")
|
| 41 |
```
|
| 42 |
```sh
|
|
|
|
|
|
|
| 43 |
def prediction(news):
|
| 44 |
news=[news]
|
| 45 |
indices=tokenizer.batch_encode_plus(
|
|
|
|
| 28 |
|
| 29 |
```sh
|
| 30 |
import torch
|
|
|
|
| 31 |
|
| 32 |
if torch.cuda.is_available():
|
| 33 |
device = torch.device("cuda")
|
|
|
|
| 39 |
device = torch.device("cpu")
|
| 40 |
```
|
| 41 |
```sh
|
| 42 |
+
import numpy as np
|
| 43 |
+
|
| 44 |
def prediction(news):
|
| 45 |
news=[news]
|
| 46 |
indices=tokenizer.batch_encode_plus(
|