Spaces:
Sleeping
Sleeping
HaggiVaggi
commited on
Commit
•
d84327f
1
Parent(s):
38006f6
Update app.py
Browse files
app.py
CHANGED
@@ -67,7 +67,7 @@ if page == "какая-то еще":
|
|
67 |
embeddings = pd.read_pickle('embeddings.pkl')
|
68 |
user_input = st.text_area('Введите описание фильма')
|
69 |
input_embedding = encode_description(user_input)
|
70 |
-
embeddings_tensor = torch.stack([torch.
|
71 |
|
72 |
|
73 |
# Рассчитайте косинусное сходство
|
|
|
67 |
embeddings = pd.read_pickle('embeddings.pkl')
|
68 |
user_input = st.text_area('Введите описание фильма')
|
69 |
input_embedding = encode_description(user_input)
|
70 |
+
embeddings_tensor = torch.stack([torch.Tensor(eval(embedding_str[2:-2])) for embedding_str in df2['description_embedding']]).numpy()
|
71 |
|
72 |
|
73 |
# Рассчитайте косинусное сходство
|