Update app.py

app.py

@@ -2,6 +2,7 @@ import transformers
 import numpy as np
 import torch
 import streamlit as st
+import re
 
 from transformers import GPT2Tokenizer, GPT2LMHeadModel
 from transformers import pipeline
@@ -54,7 +55,7 @@ if st.button("Поехали", help="Может занять какое-то в
             prompt_text + text[len(tokenizer.decode(encoded_prompt[0], clean_up_tokenization_spaces=True)) :]
         )
         
-        
+        re.sub(r'([А-Я])', r'_\n', total_sequence)
     
         #generated_sequences.append(total_sequence)
         st.write(total_sequence)