Update README.md
Browse files
README.md
CHANGED
@@ -34,8 +34,11 @@ The model leverages the BertForSequenceClassification architecture, It has been
|
|
34 |
## Example
|
35 |
|
36 |
```from transformers import AutoModelForSequenceClassification, TFAutoModelForSequenceClassification```
|
|
|
37 |
```from transformers import AutoTokenizer```
|
|
|
38 |
```import numpy as np```
|
|
|
39 |
```from scipy.special import expit```
|
40 |
|
41 |
|
@@ -55,7 +58,9 @@ The model leverages the BertForSequenceClassification architecture, It has been
|
|
55 |
|
56 |
|
57 |
```for i in range(len(predictions)):```
|
|
|
58 |
```if predictions[i]:```
|
|
|
59 |
```print(class_mapping[i])```
|
60 |
|
61 |
## Output:
|
|
|
34 |
## Example
|
35 |
|
36 |
```from transformers import AutoModelForSequenceClassification, TFAutoModelForSequenceClassification```
|
37 |
+
|
38 |
```from transformers import AutoTokenizer```
|
39 |
+
|
40 |
```import numpy as np```
|
41 |
+
|
42 |
```from scipy.special import expit```
|
43 |
|
44 |
|
|
|
58 |
|
59 |
|
60 |
```for i in range(len(predictions)):```
|
61 |
+
|
62 |
```if predictions[i]:```
|
63 |
+
|
64 |
```print(class_mapping[i])```
|
65 |
|
66 |
## Output:
|