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