minor change

reranker.py

@@ -37,16 +37,15 @@ class Reranker:
             inputs[key] = inputs[key].to(self.model.device)
         return inputs
 
-    @torch.no_grad()
-    def compute_logits(self, query, documents):
+    @torch.no_grad
+    def compute_logits(self,query,documents):
         token_false_id = self.tokenizer.convert_tokens_to_ids("no")
         token_true_id = self.tokenizer.convert_tokens_to_ids("yes")
 
         task = 'Na osnovu datog upita, vrati najrelevantije rezultate koje odgovaraju upitu'
-
         pairs = [self.format_instruction(task, query, doc) for doc in documents]
         inputs = self.process_inputs(pairs)
-    
+        
         batch_scores = self.model(**inputs).logits[:, -1, :]
         true_vector = batch_scores[:, token_true_id]
         false_vector = batch_scores[:, token_false_id]
@@ -56,4 +55,6 @@ class Reranker:
 
         results = list(zip(scores, documents))
         results.sort(key=lambda x: x[0], reverse=True)
-        return results[:10]
+        top_10 = results[:10]
+
+        return top_10