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 \
|