Update app.py

app.py

@@ -106,8 +106,10 @@ class CrossEncoderModel:
     def rerank(self, query: str, documents: List[str], top_k: int, truncation: bool) -> List[DocumentScore]:
         if not documents:
             raise ValueError("候选文档不能为空")
-        if top_k <= 0 or top_k > len(documents):
-            raise ValueError(f"top_k 需在 1~{len(documents)} 之间")
+        if top_k <= 0:
+            raise ValueError("top_k 必须为正整数")
+        # 自动将 top_k 限制为文档数量（避免超出）
+        top_k = min(top_k, len(documents))
         doc_scores = []
         for doc in documents:
             inputs = self.tokenizer(