Spaces:

MARI-posa
/

FindMyBook

Runtime error

MARI-posa commited on Jun 16, 2023

Commit

34d088a

1 Parent(s): 6acbd69

Update stri.py

Files changed (1) hide show

stri.py CHANGED Viewed

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

 if st.button('Сгенерировать'):
     with open("book_embeddingsN.pkl", "rb") as f:
+        book_embeddings = torch.load(f, map_location=torch.device('cpu'))#
         #book_embeddings = pickle.load(f)
     query_tokens = tokenizer.encode_plus(