Update README.md

README.md

@@ -21,8 +21,8 @@ def rerank(question, passages, normalize_scores=True):
     templates = [f"Query: {question}\nSentence: {passage}" for passage in passages]
     encoded_inputs = reranker_tokenizer.encode_batch(templates)
 
-    # Convert to lists and truncate sequences to max length (512)
-    input_ids = [enc.ids[:512] for enc in encoded_inputs]  # Truncate here
+    # Convert to lists and truncate sequences to max length (32768)
+    input_ids = [enc.ids[:32768] for enc in encoded_inputs]  # Truncate here
     attention_mask = [[1] * len(ids) for ids in input_ids]
 
     # Find max length in batch