Spaces:
Sleeping
Sleeping
Commit
•
51c3be1
1
Parent(s):
d84327f
Update app.py
Browse files
app.py
CHANGED
@@ -7,6 +7,7 @@ from transformers import AutoTokenizer, AutoModel
|
|
7 |
from sentence_transformers import SentenceTransformer, InputExample, losses
|
8 |
from torch.utils.data import DataLoader
|
9 |
import torch
|
|
|
10 |
|
11 |
|
12 |
|
@@ -67,7 +68,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.Tensor(
|
71 |
|
72 |
|
73 |
# Рассчитайте косинусное сходство
|
|
|
7 |
from sentence_transformers import SentenceTransformer, InputExample, losses
|
8 |
from torch.utils.data import DataLoader
|
9 |
import torch
|
10 |
+
import ast
|
11 |
|
12 |
|
13 |
|
|
|
68 |
embeddings = pd.read_pickle('embeddings.pkl')
|
69 |
user_input = st.text_area('Введите описание фильма')
|
70 |
input_embedding = encode_description(user_input)
|
71 |
+
embeddings_tensor = torch.stack([torch.Tensor(ast.literal_eval(embedding_str)) for embedding_str in df2['description_embedding']]).numpy()
|
72 |
|
73 |
|
74 |
# Рассчитайте косинусное сходство
|