Update stri.py

stri.py

@@ -46,7 +46,8 @@ query = st.text_input("Введите запрос")
 
 if st.button('Сгенерировать'):
     with open("book_embeddingsN.pkl", "rb") as f:
-        book_embeddings = pickle.load(f)
+        book_embeddings = pickle.load(f, map_location=torch.device('cpu'))#
+        #book_embeddings = pickle.load(f)
 
     query_tokens = tokenizer.encode_plus(
             query,
@@ -65,7 +66,7 @@ if st.button('Сгенерировать'):
     # Вычисление косинусного расстояния между эмбеддингом запроса и каждой аннотацией
     cosine_similarities = torch.nn.functional.cosine_similarity(
         query_embedding.squeeze(0),
-        torch.stack(book_embeddings.cpu())
+        torch.stack(book_embeddings)
     )
 
     cosine_similarities = cosine_similarities.numpy()