Update app.py

app.py

@@ -12,12 +12,13 @@ max_len = st.text_area("Максимальная длина")
 text = st.text_area("Тема сочинения")
 # ^-- показать текстовое поле. В поле text лежит строка, которая находится там в данный момент
 thesis = "<s>Тема:" + "«" + text + "»." + "\nСочинение: " 
-begin_index = len(thesis)
 
 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>")
     
     st.markdown(f"Сочинение:\n")
-    st.markdown(generated[0][begin_index:])
+    st.markdown(generated[0][begin_index:end_index])
 # выводим результаты модели в текстовое поле, на потеху пользователю