Spaces:
Runtime error

dzenzzz commited on
Commit
de9714c
·
1 Parent(s): c6b5790

minor change

Browse files
Files changed (1) hide show
  1. reranker.py +1 -1
reranker.py CHANGED
@@ -7,7 +7,7 @@ class Reranker:
7
  self.tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen3-Reranker-4B", padding_side='left')
8
  self.model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen3-Reranker-4B").eval()
9
 
10
- def format_instruction(instruction, query, doc):
11
  if instruction is None:
12
  instruction = 'Given a web search query, retrieve relevant passages that answer the query'
13
  output = "<Instruct>: {instruction}\n<Query>: {query}\n<Document>: {doc}".format(instruction=instruction,query=query, doc=doc)
 
7
  self.tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen3-Reranker-4B", padding_side='left')
8
  self.model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen3-Reranker-4B").eval()
9
 
10
+ def format_instruction(self,instruction, query, doc):
11
  if instruction is None:
12
  instruction = 'Given a web search query, retrieve relevant passages that answer the query'
13
  output = "<Instruct>: {instruction}\n<Query>: {query}\n<Document>: {doc}".format(instruction=instruction,query=query, doc=doc)