Update stri.py

stri.py

@@ -65,7 +65,7 @@ if st.button('Сгенерировать'):
     # Вычисление косинусного расстояния между эмбеддингом запроса и каждой аннотацией
     cosine_similarities = torch.nn.functional.cosine_similarity(
         query_embedding.squeeze(0),
-        torch.stack(book_embeddings).cpu()
+        torch.stack(book_embeddings.cpu())
     )
 
     cosine_similarities = cosine_similarities.numpy()