Spaces:
Runtime error
Runtime error
Update stri.py
Browse files
stri.py
CHANGED
|
@@ -57,7 +57,7 @@ for inputs, attention_masks in zip(input_ids, attention_mask):
|
|
| 57 |
book_embeddings.append(np.squeeze(book_embedding))
|
| 58 |
|
| 59 |
# Определение запроса пользователя
|
| 60 |
-
query = "
|
| 61 |
query_tokens = tokenizer.encode(query, add_special_tokens=True, \
|
| 62 |
truncation=True, max_length=max_len)
|
| 63 |
|
|
|
|
| 57 |
book_embeddings.append(np.squeeze(book_embedding))
|
| 58 |
|
| 59 |
# Определение запроса пользователя
|
| 60 |
+
query = st.text_input("Введите запрос")
|
| 61 |
query_tokens = tokenizer.encode(query, add_special_tokens=True, \
|
| 62 |
truncation=True, max_length=max_len)
|
| 63 |
|