HaggiVaggi commited on
Commit
51c3be1
1 Parent(s): d84327f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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(eval(embedding_str[2:-2])) for embedding_str in df2['description_embedding']]).numpy()
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
  # Рассчитайте косинусное сходство