Update README.md
Browse files
README.md
CHANGED
|
@@ -26,7 +26,7 @@ from torch.nn.functional import softmax, sigmoid
|
|
| 26 |
|
| 27 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 28 |
|
| 29 |
-
model_name="ragarwal/deberta-v3-base-nli-mixer"
|
| 30 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
| 31 |
model = AutoModelForSequenceClassification.from_pretrained(model_name)
|
| 32 |
|
|
@@ -52,7 +52,7 @@ In Sentence-Transformers
|
|
| 52 |
```python
|
| 53 |
from sentence_transformers import CrossEncoder
|
| 54 |
|
| 55 |
-
model_name="ragarwal/deberta-v3-base-nli-mixer"
|
| 56 |
model = CrossEncoder(model_name, max_length=256)
|
| 57 |
|
| 58 |
sentence = "During its monthly call, the National Oceanic and Atmospheric Administration warned of \
|
|
|
|
| 26 |
|
| 27 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 28 |
|
| 29 |
+
model_name="ragarwal/deberta-v3-base-nli-mixer-binary"
|
| 30 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
| 31 |
model = AutoModelForSequenceClassification.from_pretrained(model_name)
|
| 32 |
|
|
|
|
| 52 |
```python
|
| 53 |
from sentence_transformers import CrossEncoder
|
| 54 |
|
| 55 |
+
model_name="ragarwal/deberta-v3-base-nli-mixer-binary"
|
| 56 |
model = CrossEncoder(model_name, max_length=256)
|
| 57 |
|
| 58 |
sentence = "During its monthly call, the National Oceanic and Atmospheric Administration warned of \
|