bankholdup commited on
Commit
9104177
1 Parent(s): ed2cd9d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -21,8 +21,6 @@ title = st.title("Загрузка модели")
21
  tokenizer, model = load_model()
22
  title.title("ruGPT3 Song Writer")
23
  context = st.text_input("Введите начало песни", "Как дела? Как дела? Это новый кадиллак")
24
- top_k = st.slider("top_k", 0, 100, 50)
25
- top_p = st.slider("top_p", 0.0, 10.0, 0.95)
26
  temperature= st.slider("temperature", 0.0, 10.0, 0.95)
27
 
28
  if st.button("Поехали", help="Может занять какое-то время"):
@@ -39,8 +37,8 @@ if st.button("Поехали", help="Может занять какое-то в
39
  input_ids=encoded_prompt,
40
  max_length=200 + len(encoded_prompt[0]),
41
  temperature=temperature,
42
- top_k=top_k,
43
- top_p=top_p,
44
  repetition_penalty=1.0,
45
  do_sample=True,
46
  num_return_sequences=1,
 
21
  tokenizer, model = load_model()
22
  title.title("ruGPT3 Song Writer")
23
  context = st.text_input("Введите начало песни", "Как дела? Как дела? Это новый кадиллак")
 
 
24
  temperature= st.slider("temperature", 0.0, 10.0, 0.95)
25
 
26
  if st.button("Поехали", help="Может занять какое-то время"):
 
37
  input_ids=encoded_prompt,
38
  max_length=200 + len(encoded_prompt[0]),
39
  temperature=temperature,
40
+ top_k=50,
41
+ top_p=0.95,
42
  repetition_penalty=1.0,
43
  do_sample=True,
44
  num_return_sequences=1,