Spaces:
Build error
Build error
bankholdup
commited on
Commit
•
9104177
1
Parent(s):
ed2cd9d
Update app.py
Browse files
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=
|
43 |
-
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,
|