Spaces:
Sleeping
Sleeping
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="Чем подробнее будет ваше описание, тем точнее мы сможем подобрать для вас фильм 🤗'")
|