Update README.md
Browse files
README.md
CHANGED
@@ -75,7 +75,7 @@ To ensure reproducibility:
|
|
75 |
```python
|
76 |
from sentence_transformers import CrossEncoder
|
77 |
|
78 |
-
model_name = "e5-small-v2-nli"
|
79 |
model = CrossEncoder(model_name)
|
80 |
scores = model.predict(
|
81 |
[
|
@@ -98,7 +98,7 @@ print(labels)
|
|
98 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
99 |
import torch
|
100 |
|
101 |
-
model_name = "e5-small-v2-nli"
|
102 |
model = AutoModelForSequenceClassification.from_pretrained(model_name)
|
103 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
104 |
|
|
|
75 |
```python
|
76 |
from sentence_transformers import CrossEncoder
|
77 |
|
78 |
+
model_name = "agentlans/e5-small-v2-nli"
|
79 |
model = CrossEncoder(model_name)
|
80 |
scores = model.predict(
|
81 |
[
|
|
|
98 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
99 |
import torch
|
100 |
|
101 |
+
model_name = "agentlans/e5-small-v2-nli"
|
102 |
model = AutoModelForSequenceClassification.from_pretrained(model_name)
|
103 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
104 |
|