import streamlit as st
from Generate_text import generate, load_tokenizer_and_model

st.markdown("### Hello, world!")
st.markdown("<img width=200px src='https://rozetked.me/images/uploads/dwoilp3BVjlE.jpg'>", unsafe_allow_html=True)
# ^-- можно показывать пользователю текст, картинки, ограниченное подмножество html - всё как в jupyter

text = st.text_area("TEXT HERE")
# ^-- показать текстовое поле. В поле text лежит строка, которая находится там в данный момент
device = "cpu"
tok, model = load_tokenizer_and_model("models/essays", device)
generated = generate(model, tok, "<s>Тема: «Как не застрелиться?». (В.Т. Белинский)\nСочинение: ", device, num_beams=10)

st.markdown(f"Hello!\n")
st.markdown(generated[0])
# выводим результаты модели в текстовое поле, на потеху пользователю