Spaces:
Runtime error
Runtime error
minor change
Browse files- 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)
|