Update app.py

app.py

@@ -25,12 +25,12 @@ context = st.text_input("Введите начало песни", "Как дел
 if st.button("Поехали", help="Может занять какое-то время"):
     generated_sequences = []
     set_seed()
-    st.write("Генерируем")
+    st.write("Генерируем...")
     prompt_text = f"{context}"
     encoded_prompt = tokenizer.encode(prompt_text, add_special_tokens=False, return_tensors="pt")
     output_sequences = model.generate(
             input_ids=encoded_prompt,
-            max_length=250 + len(encoded_prompt[0]),
+            max_length=200 + len(encoded_prompt[0]),
             temperature=1.0,
             top_k=50,
             top_p=0.95,
@@ -44,14 +44,11 @@ if st.button("Поехали", help="Может занять какое-то в
     for generated_sequence_idx, generated_sequence in enumerate(output_sequences):
         generated_sequence = generated_sequence.tolist()
         text = tokenizer.decode(generated_sequence, clean_up_tokenization_spaces=True)
-    
-        #text = text[: text.find("/n") if "/n" else None]
 
         total_sequence = (
             prompt_text + text[len(tokenizer.decode(encoded_prompt[0], clean_up_tokenization_spaces=True)) :]
         )
-    
-        #generated_sequences.append(total_sequence)
+
         splits = total_sequence.splitlines()
         for line in range(len(splits)-1):
             if "[" in splits[line]: