ragarwal commited on
Commit
338b199
1 Parent(s): c08b06f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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 \