Update pages/Подбор фильмов по описанию✏️🔍.py

pages/Подбор фильмов по описанию✏️🔍.py

@@ -24,10 +24,10 @@ def load_model():
 tokenizer = AutoTokenizer.from_pretrained("DeepPavlov/rubert-base-cased-sentence")
 
 @st.cache_data
-def encode_description(description, tokenizer, model):
-    tokens = tokenizer(description, return_tensors="pt")
+def encode_description(_tokenizer, _model, description):
+    tokens = _tokenizer(description, return_tensors="pt")
     with torch.no_grad():
-        outputs = model(**tokens)
+        outputs = _model(**tokens)
     embeddings = outputs.last_hidden_state.mean(dim=1)
     return embeddings.cpu().numpy().astype('float32')