MARI-posa commited on
Commit
8e3efa7
1 Parent(s): 5bbc734

Update stri.py

Browse files
Files changed (1) hide show
  1. stri.py +1 -1
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