Update app.py

app.py

@@ -1,19 +1,20 @@
 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)
+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
 
-text = st.text_area("TEXT HERE")
+max_len = int(st.text_area("Максимальная длина"))
+text = st.text_area("Тема сочинения")
 # ^-- показать текстовое поле. В поле text лежит строка, которая находится там в данный момент
 device = "cpu"
 thesis = "<s>Тема:" + "«" + text + "»." + "\nСочинение: " 
 begin_index = len(thesis)
 
 tok, model = load_tokenizer_and_model("models/essays", device)
-generated = generate(model, tok, thesis, device, num_beams=10)
+generated = generate(model, tok, thesis, device, max_length=max_len, num_beams=10)
 
-st.markdown(f"Hello!\n")
+st.markdown(f"Сочинение:\n")
 st.markdown(generated[0][begin_index:])
 # выводим результаты модели в текстовое поле, на потеху пользователю