Update app.py

app.py

@@ -27,7 +27,6 @@ title.title("ruGPT3 Song Writer")
 context = st.text_input("Введите начало песни", "Как дела? Как дела? Это новый кадиллак")
 
 if st.button("Поехали", help="Может занять какое-то время"):
-    st.title(f"{context}")
     prefix_text = f"{context}"
     encoded_prompt = tokenizer.encode(prefix_text, add_special_tokens=False, return_tensors="pt")
     output_sequences = model.generate(
@@ -53,7 +52,7 @@ if st.button("Поехали", help="Может занять какое-то в
         text = tokenizer.decode(generated_sequence, clean_up_tokenization_spaces=True)
     
         # Remove all text after the stop token
-        text = text[: text.find(args.stop_token) if args.stop_token else None]
+        text = text[: text.find("</s>") if "</s>" else None]
     
         # Add the prompt at the beginning of the sequence. Remove the excess text that was used for pre-processing
         total_sequence = (