HaggiVaggi commited on
Commit
600a6d6
·
1 Parent(s): 071f0b8

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

Browse files
pages/Подбор фильмов по описанию✏️🔍.py CHANGED
@@ -25,8 +25,6 @@ def load_tokenizer_and_model():
25
  tokenizer = AutoTokenizer.from_pretrained("DeepPavlov/rubert-base-cased-sentence")
26
  model = AutoModel.from_pretrained("DeepPavlov/rubert-base-cased-sentence")
27
  return tokenizer, model
28
-
29
- tokenizer, model = load_tokenizer_and_model()
30
 
31
  @st.cache_resource
32
  def encode_description(description, tokenizer, model):
@@ -38,6 +36,9 @@ def encode_description(description, tokenizer, model):
38
 
39
  st.header("Подбор фильмов по описанию ✏️🔍")
40
 
 
 
 
41
 
42
  # Пользовательский ввод
43
  user_input = st.text_input("Введите описание фильма:", value="", help="Чем подробнее будет ваше описание, тем точнее мы сможем подобрать для вас фильм 🤗'")
 
25
  tokenizer = AutoTokenizer.from_pretrained("DeepPavlov/rubert-base-cased-sentence")
26
  model = AutoModel.from_pretrained("DeepPavlov/rubert-base-cased-sentence")
27
  return tokenizer, model
 
 
28
 
29
  @st.cache_resource
30
  def encode_description(description, tokenizer, model):
 
36
 
37
  st.header("Подбор фильмов по описанию ✏️🔍")
38
 
39
+ df = load_data('data/final_data.csv')
40
+ embeddings_array, index = embedding_and_index()
41
+ tokenizer, model = load_tokenizer_and_model()
42
 
43
  # Пользовательский ввод
44
  user_input = st.text_input("Введите описание фильма:", value="", help="Чем подробнее будет ваше описание, тем точнее мы сможем подобрать для вас фильм 🤗'")