diff --git a/backend/app/xinference/base.py b/backend/app/xinference/base.py index 374c2ca..f256ec8 100644 --- a/backend/app/xinference/base.py +++ b/backend/app/xinference/base.py @@ -224,8 +224,8 @@ class XinferenceRerank(BaseNodePostprocessor): if self.top_n is not None: if len(new_nodes) > self.top_n: - for index in new_nodes[5:-1]: - new_nodes.remove(-1) + for index in new_nodes[self.top_n:-1]: + new_nodes.remove(index) event.on_end(payload={EventPayload.NODES: new_nodes})