Spaces:
Runtime error
Runtime error
Update stri.py
Browse files
stri.py
CHANGED
@@ -45,10 +45,10 @@ length = 512
|
|
45 |
query = st.text_input("Введите запрос")
|
46 |
|
47 |
if st.button('Сгенерировать'):
|
48 |
-
with open("
|
49 |
book_embeddings = pickle.load(f)
|
50 |
|
51 |
-
book_embeddings = torch.tensor(book_embeddings, device=torch.device('cpu'))
|
52 |
|
53 |
|
54 |
#if st.button('Сгенерировать'):
|
|
|
45 |
query = st.text_input("Введите запрос")
|
46 |
|
47 |
if st.button('Сгенерировать'):
|
48 |
+
with open(" book_embeddings256xxx.pkl", "rb") as f:
|
49 |
book_embeddings = pickle.load(f)
|
50 |
|
51 |
+
#book_embeddings = torch.tensor(book_embeddings, device=torch.device('cpu'))
|
52 |
|
53 |
|
54 |
#if st.button('Сгенерировать'):
|