Update stri.py

stri.py

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