Update app.py

app.py

@@ -22,9 +22,9 @@ title = st.title("Загрузка модели")
 tokenizer, model = load_model()
 title.title("ruGPT3 Song Writer")
 context = st.text_input("Введите начало песни", "Как дела? Как дела? Это новый кадиллак")
-generated_sequences = []
 
 if st.button("Поехали", help="Может занять какое-то время"):
+    generated_sequences = []
     set_seed()
     st.title("Генерируем")
     prompt_text = f"{context}"
@@ -41,21 +41,20 @@ if st.button("Поехали", help="Может занять какое-то в
         )
     if len(output_sequences.shape) > 2:
         output_sequences.squeeze_()
+
+    st.title("")
     
     for generated_sequence_idx, generated_sequence in enumerate(output_sequences):
-        print("ruGPT:".format(generated_sequence_idx + 1))
         generated_sequence = generated_sequence.tolist()
-
         text = tokenizer.decode(generated_sequence, clean_up_tokenization_spaces=True)
-        st.write(text)
     
-       # text = text[: text.find("</s>") if "</s>" else None]
+        text = text[: text.find("</s>") if "</s>" else None]
 
-        #total_sequence = (
-        #    prompt_text + text[len(tokenizer.decode(encoded_prompt[0], clean_up_tokenization_spaces=True)) :]
-        #)
+        total_sequence = (
+            prompt_text + text[len(tokenizer.decode(encoded_prompt[0], clean_up_tokenization_spaces=True)) :]
+        )
     
         # generated_sequences.append(total_sequence)
         # os.system('clear')
-        st.title("")
-        #st.write(total_sequence)
+
+        st.write(total_sequence)