Spaces:
Runtime error
Runtime error
File size: 1,359 Bytes
1ea64bd 7eac112 1ea64bd a418b39 9d6d875 1ea64bd 4cebd63 9d6d875 1ea64bd fe1596b d306847 9ba07dc d306847 a418b39 4cebd63 a418b39 4cebd63 1ea64bd |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
import streamlit as st
from Generate_text import generate, load_tokenizer_and_model
tok, model = load_tokenizer_and_model("models/essays", device)
st.markdown("### Это приложение представляет из себя примтивный генератор сочинений. ")
st.markdown("<img width=200px src='https://ps-static.cdn-tinkoff.ru/static/ai-pushkin/portrait-2021-12-10-12-24-56.png'>", unsafe_allow_html=True)
# ^-- можно показывать пользователю текст, картинки, ограниченное подмножество html - всё как в jupyter
max_len = st.text_area("Максимальная длина")
text = st.text_area("Тема сочинения")
# ^-- показать текстовое поле. В поле text лежит строка, которая находится там в данный момент
device = "cpu"
thesis = "<s>Тема:" + "«" + text + "»." + "\nСочинение: "
begin_index = len(thesis)
if max_len != "":
max_len = int(max_len)
generated = generate(model, tok, thesis, device, max_length=max_len, num_beams=10)
st.markdown(f"Сочинение:\n")
st.markdown(generated[0][begin_index:])
# выводим результаты модели в текстовое поле, на потеху пользователю |