Update stri.py

stri.py

@@ -71,9 +71,13 @@ with torch.no_grad():
     query_embedding = query_embedding[0][:, 0, :].detach().cpu().numpy()
 
 # Вычисление косинусного расстояния между эмбеддингом запроса и каждой аннотацией
+#cosine_similarities = torch.nn.functional.cosine_similarity(
+#    query_embedding.squeeze(0),
+#    torch.stack(book_embeddings)
+#)
 cosine_similarities = torch.nn.functional.cosine_similarity(
-    query_embedding.squeeze(0),
-    torch.stack(book_embeddings)
+    torch.tensor(query_embedding.squeeze(0)),
+    torch.stack([torch.tensor(embedding) for embedding in book_embeddings])
 )
 
 cosine_similarities = cosine_similarities.numpy()