HaggiVaggi commited on
Commit
d84327f
1 Parent(s): 38006f6

Update app.py

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