Upload reranker.py with huggingface_hub

reranker.py

@@ -4,8 +4,9 @@ from transformers import AutoModel, AutoTokenizer, AutoModelForCausalLM
 class Reranker:
     
     def __init__(self):
-        self.tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen3-Reranker-4B", padding_side='left')
-        self.model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen3-Reranker-4B").eval()
+        # Koristi 0.6B model umesto 4B zbog manje memorije
+        self.tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen3-Reranker-0.6B", padding_side='left')
+        self.model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen3-Reranker-0.6B").eval()
 
     def format_instruction(self,instruction, query, doc):
         if instruction is None: