Hovhannes commited on
Commit
7ab9cc5
Β·
verified Β·
1 Parent(s): 275d886

Update movie_recommender.py

Browse files
Files changed (1) hide show
  1. 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-4B", padding_side='left')
94
- self.reranker_model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen3-Reranker-4B").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
 
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