cmarkea
/

bloomz-3b-reranking

text-classification

Model card Files Files and versions

Cyrile commited on Apr 11, 2024

Commit

05e1d3f

·

verified ·

1 Parent(s): 5b26c69

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -95,7 +95,7 @@ from transformers import pipeline
 reranker = pipeline(
     task='feature-extraction',
-    model='cmarkea/bloomz-3b-retriever',
     top_k=None
 )

 reranker = pipeline(
     task='feature-extraction',
+    model='cmarkea/bloomz-3b-reranking',
     top_k=None
 )