Update pipeline tag and add library name

README.md

@@ -1,15 +1,16 @@
 ---
-license: mit
-datasets:
-- jhu-clsp/rank1-training-data
 base_model:
 - Qwen/Qwen2.5-32B
-pipeline_tag: text-generation
+datasets:
+- jhu-clsp/rank1-training-data
+language:
+- en
+license: mit
+pipeline_tag: feature-extraction
+library_name: transformers
 tags:
 - reranker
 - retrieval
-language:
-- en
 ---
 
 # rank1-32b: Test-Time Compute for Reranking in Information Retrieval
@@ -141,7 +142,7 @@ Please see the Github for detailed installation instructions.
 
 ## MTEB Integration
 
-rank1 is compatible with the [MTEB benchmarking framework](https://github.com/embeddings-benchmark/mteb):
+rank1 is compatible with the [MTEB benchmarking framework](https://github.com/embeddings-benchmark/mteb):\
 
 ```python
 from mteb import MTEB
@@ -149,7 +150,7 @@ from rank1 import rank1  # From the official repo
 
 # Initialize the model
 model = rank1(
-    model_name_or_path="jhu-clsp/rank1-32b",
+    model_name_or_path="jhu-clsp/rank1-7b",
     num_gpus=1,
     device="cuda"
 )
@@ -177,4 +178,4 @@ If you use rank1 in your research, please cite our work:
 
 ## License
 
-[MIT License](https://github.com/orionw/rank1/blob/main/LICENSE) 
+[MIT License](https://github.com/orionw/rank1/blob/main/LICENSE)