Update README.md

README.md

@@ -58,17 +58,16 @@ We have two models:
 1. Install mxbai-reranker
 
 ```bash
-pip install git+https://github.com/mixedbread-ai/mxbai-reranker.git
+pip install mxbai-rerank
 ```
 
 
 2. Inference
 
 ```python
-from mxbai_reranker.v2 import MxbaiReranker
+from mxbai_rerank import MxbaiRerankV2
 
-# pass attn_implementation="flash_attention_2" to use flash attention
-model = MxbaiReranker.from_pretrained("mixedbread-ai/mxbai-rerank-base-v2").to("cuda")
+model = MxbaiRerankV2("mixedbread-ai/mxbai-rerank-base-v2")
 
 query = "Who wrote 'To Kill a Mockingbird'?"
 documents = [