fix: add missing module

README.md

@@ -50,48 +50,16 @@ curl https://api.jina.ai/v1/rerank \
 }'
 ```
 
-2. Alternatively, you can use the latest version of the `sentence-transformers>=0.27.0` library. You can install it via pip:
-
-```bash
-pip install -U sentence-transformers
-```
-
-Then, you can use the following code to interact with the model:
-
-```python
-from sentence_transformers import CrossEncoder
-
-# Load the model, here we use our base sized model
-model = CrossEncoder("jinaai/jina-reranker-v2-base-multilingual", num_labels=1, trust_remote_code=True)
-
-
-# Example query and documents
-query = "Organic skincare products for sensitive skin"
-documents = [
-    "Eco-friendly kitchenware for modern homes",
-    "Biodegradable cleaning supplies for eco-conscious consumers",
-    "Organic cotton baby clothes for sensitive skin",
-    "Natural organic skincare range for sensitive skin",
-    "Tech gadgets for smart homes: 2024 edition",
-    "Sustainable gardening tools and compost solutions",
-    "Sensitive skin-friendly facial cleansers and toners",
-    "Organic food wraps and storage solutions",
-    "All-natural pet food for dogs with allergies",
-    "Yoga mats made from recycled materials"
-]
-
-results = model.rank(query, documents, return_documents=True, top_k=3)
-```
-
-3. You can also use the `transformers` library to interact with the model programmatically.
+2. You can also use the `transformers` library to interact with the model programmatically.
 
 ```python
 !pip install transformers
 from transformers import AutoModelForSequenceClassification
 
 model = AutoModelForSequenceClassification.from_pretrained(
-    'jinaai/jina-reranker-v2-base-multilingual', trust_remote_code=True
+    'jinaai/jina-reranker-v2-base-multilingual', trust_remote_code=True,
 )
+model.to('cuda')
 
 # Example query and documents
 query = "Organic skincare products for sensitive skin"