Update app.py

app.py

@@ -21,20 +21,26 @@ title = st.title("Загрузка модели")
 tokenizer, model = load_model()
 title.title("ruGPT3 Song Writer")
 context = st.text_input("Введите начало песни", "Как дела? Как дела? Это новый кадиллак")
+top_k = st.slider("top_k", 0, 100, 50)
+top_p = st.slider("top_p", 0.0, 10.0, 0.95)
+temperature= st.slider("top_p", 0.0, 10.0, 0.95)
 
 if st.button("Поехали", help="Может занять какое-то время"):
     generated_sequences = []
     set_seed()
     st.write("Генерируем...")
+    st.write("top_k = {}".format(top_k))
+    st.write("top_p = {}".format(top_p))
+    st.write("temperature = {}".format(temperature))
     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=200 + len(encoded_prompt[0]),
-            temperature=1.0,
-            top_k=50,
-            top_p=0.95,
+            temperature=temperature,
+            top_k=top_k,
+            top_p=top_p,
             repetition_penalty=1.0,
             do_sample=True,
             num_return_sequences=1,