bankholdup commited on
Commit
91c7bac
1 Parent(s): 1a8562f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -20,8 +20,8 @@ def set_seed(rng=100000):
20
  title = st.title("Загрузка модели")
21
  tokenizer, model = load_model()
22
  title.title("ruGPT3 Song Writer")
23
- context = st.text_input("Введите начало песни", "Как дела? Как дела? Это новый кадиллак")
24
- temperature= st.slider("temperature", 0.0, 2.5, 0.95)
25
 
26
  if st.button("Поехали", help="Может занять какое-то время"):
27
  generated_sequences = []
@@ -34,7 +34,7 @@ if st.button("Поехали", help="Может занять какое-то в
34
  output_sequences = model.generate(
35
  input_ids=encoded_prompt,
36
  max_length=200 + len(encoded_prompt[0]),
37
- temperature=0.95,
38
  top_k=50,
39
  top_p=0.95,
40
  repetition_penalty=1.0,
 
20
  title = st.title("Загрузка модели")
21
  tokenizer, model = load_model()
22
  title.title("ruGPT3 Song Writer")
23
+ context = st.text_input("Введите начало песни", "Батя возвращается трезвым, в руке буханка")
24
+ temperature= st.slider("temperature (чем выше, тем текст безумнее, чем ниже, тем ближе к исходным данным)", 0.0, 2.5, 0.95)
25
 
26
  if st.button("Поехали", help="Может занять какое-то время"):
27
  generated_sequences = []
 
34
  output_sequences = model.generate(
35
  input_ids=encoded_prompt,
36
  max_length=200 + len(encoded_prompt[0]),
37
+ temperature=temperature,
38
  top_k=50,
39
  top_p=0.95,
40
  repetition_penalty=1.0,