Update app.py

app.py

@@ -21,14 +21,16 @@ def set_seed(args):
     if args.n_gpu > 0:
         torch.cuda.manual_seed_all(rd)
 
-title = st.title("Loading model")
+title = st.title("Загрузка модели")
 tokenizer, model = load_model()
 title.title("ruGPT3 Song Writer")
 context = st.text_input("Введите начало песни", "Как дела? Как дела? Это новый кадиллак")
+generated_sequences = []
 
 if st.button("Поехали", help="Может занять какое-то время"):
-    prefix_text = f"{context}"
-    encoded_prompt = tokenizer.encode(prefix_text, add_special_tokens=False, return_tensors="pt")
+    st.title(f"Новая топовая песня")
+    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=200 + len(encoded_prompt[0]),