Spaces:
Runtime error
Runtime error
Update movie_recommender.py
Browse files- movie_recommender.py +2 -2
movie_recommender.py
CHANGED
|
@@ -90,8 +90,8 @@ class MovieRecommender:
|
|
| 90 |
|
| 91 |
# Initialize reranker components
|
| 92 |
print("Loading reranker model...")
|
| 93 |
-
self.reranker_tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen3-Reranker-
|
| 94 |
-
self.reranker_model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen3-Reranker-
|
| 95 |
self.token_false_id = self.reranker_tokenizer.convert_tokens_to_ids("no")
|
| 96 |
self.token_true_id = self.reranker_tokenizer.convert_tokens_to_ids("yes")
|
| 97 |
self.max_length = 8192
|
|
|
|
| 90 |
|
| 91 |
# Initialize reranker components
|
| 92 |
print("Loading reranker model...")
|
| 93 |
+
self.reranker_tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen3-Reranker-0.6B", padding_side='left')
|
| 94 |
+
self.reranker_model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen3-Reranker-0.6B").eval()
|
| 95 |
self.token_false_id = self.reranker_tokenizer.convert_tokens_to_ids("no")
|
| 96 |
self.token_true_id = self.reranker_tokenizer.convert_tokens_to_ids("yes")
|
| 97 |
self.max_length = 8192
|