Update stri.py

stri.py

@@ -45,7 +45,7 @@ length = 512
 query = st.text_input("Введите запрос")
 
 if st.button('Сгенерировать'):
-    with open("book_embeddings256xxx.pkl", "rb") as f:
+    with open("book_embeddings1.pkl", "rb") as f:
         book_embeddings = pickle.load(f)
 
     #book_embeddings = torch.tensor(book_embeddings, device=torch.device('cpu'))
@@ -73,7 +73,7 @@ if st.button('Сгенерировать'):
         
     # Вычисление косинусного расстояния между эмбеддингом запроса и каждой аннотацией
     cosine_similarities = torch.nn.functional.cosine_similarity(
-        query_embedding.squeeze(0),
+        query_hidden_states.squeeze(0),
         torch.stack(book_embeddings)
     )