Update stri.py

stri.py

@@ -45,8 +45,15 @@ length = 512
 query = st.text_input("Введите запрос")
 
 if st.button('Сгенерировать'):
+    with open("book_embeddingsN.pkl", "rb") as f:
+        book_embeddings = pickle.load(f)
+
+    book_embeddings = torch.tensor(book_embeddings, device=torch.device('cpu'))
+
+
+#if st.button('Сгенерировать'):
     #with open("book_embeddingsN.pkl", "rb") as f:
-    book_embeddings = torch.load("book_embeddingsN.pkl", map_location=torch.device('cpu'))
+    #book_embeddings = torch.load("book_embeddingsN.pkl", map_location=torch.device('cpu'))
 #
         #book_embeddings = pickle.load(f)