numb3r3 commited on
Commit
081430d
1 Parent(s): 538bd1e

chore: update example code

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -75,7 +75,7 @@ Then, you can use the following code to interact with the model:
75
  from sentence_transformers import CrossEncoder
76
 
77
  # Load the model, here we use our base sized model
78
- model = CrossEncoder("jina-reranker-v1-turbo-en", num_labels=1, automodel_args={'trust_remote_code': True})
79
 
80
 
81
  # Example query and documents
@@ -103,7 +103,7 @@ results = model.rank(query, documents, return_documents=True, top_k=3)
103
  from transformers import AutoModelForSequenceClassification
104
 
105
  model = AutoModelForSequenceClassification.from_pretrained(
106
- 'jina-reranker-v1-turbo-en', num_labels=1, trust_remote_code=True
107
  )
108
 
109
  # Example query and documents
 
75
  from sentence_transformers import CrossEncoder
76
 
77
  # Load the model, here we use our base sized model
78
+ model = CrossEncoder("jinaai/jina-reranker-v1-turbo-en", num_labels=1, automodel_args={'trust_remote_code': True})
79
 
80
 
81
  # Example query and documents
 
103
  from transformers import AutoModelForSequenceClassification
104
 
105
  model = AutoModelForSequenceClassification.from_pretrained(
106
+ 'jinaai/jina-reranker-v1-turbo-en', num_labels=1, trust_remote_code=True
107
  )
108
 
109
  # Example query and documents