Update app.py

app.py

@@ -4,7 +4,8 @@ from Generate_text import generate, load_tokenizer_and_model
 device = "cpu"
 tok, model = load_tokenizer_and_model("models/essays", device)
 
-st.markdown("### Это приложение представляет из себя примтивный генератор сочинений. ")
+st.markdown("### Генератор сочинений")
+st.markdown("# Это приложение представляет из себя примитивный генератор сочинений. В качестве основной модели был взят трансформер RuGPT от Сбера https://huggingface.co/ai-forever/rugpt3small_based_on_gpt2")
 with st.columns(3)[1]:
     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
@@ -18,7 +19,7 @@ if max_len != "":
     begin_index = len(thesis)
     max_len = int(max_len)
     generated = generate(model, tok, thesis, device, max_length=max_len, num_beams=10)
-    end_index = generated.find("</s>")
+    end_index = (generated[0]).find("</s>")
     
     st.markdown(f"Сочинение:\n")
     st.markdown(generated[0][begin_index:end_index])