agentlans commited on
Commit
6da84ba
1 Parent(s): a0cabdd

Update README.md

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