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