Update app.py

app.py

@@ -2,10 +2,8 @@ import transformers
 import numpy as np
 import torch
 import streamlit as st
-import re
 
 from transformers import GPT2Tokenizer, GPT2LMHeadModel
-from transformers import pipeline
 
 @st.cache(allow_output_mutation=True)
 def load_model():
@@ -55,7 +53,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)
+        reformat_seq = ''.join(['\n'+ x if x.isupper() else x for x in total_sequence ])
     
         #generated_sequences.append(total_sequence)
         st.write(total_sequence)