Spaces:
Build error
Build error
bankholdup
commited on
Commit
•
41e6718
1
Parent(s):
73288d7
Update app.py
Browse files
app.py
CHANGED
@@ -28,7 +28,6 @@ context = st.text_input("Введите начало песни", "Как дел
|
|
28 |
generated_sequences = []
|
29 |
|
30 |
if st.button("Поехали", help="Может занять какое-то время"):
|
31 |
-
st.title(f"Новая топовая песня")
|
32 |
prompt_text = f"{context}"
|
33 |
encoded_prompt = tokenizer.encode(prompt_text, add_special_tokens=False, return_tensors="pt")
|
34 |
output_sequences = model.generate(
|
@@ -58,7 +57,7 @@ if st.button("Поехали", help="Может занять какое-то в
|
|
58 |
|
59 |
# Add the prompt at the beginning of the sequence. Remove the excess text that was used for pre-processing
|
60 |
total_sequence = (
|
61 |
-
prompt_text + text[len(tokenizer.decode(encoded_prompt[0], clean_up_tokenization_spaces=True)) :]
|
62 |
)
|
63 |
|
64 |
generated_sequences.append(total_sequence)
|
|
|
28 |
generated_sequences = []
|
29 |
|
30 |
if st.button("Поехали", help="Может занять какое-то время"):
|
|
|
31 |
prompt_text = f"{context}"
|
32 |
encoded_prompt = tokenizer.encode(prompt_text, add_special_tokens=False, return_tensors="pt")
|
33 |
output_sequences = model.generate(
|
|
|
57 |
|
58 |
# Add the prompt at the beginning of the sequence. Remove the excess text that was used for pre-processing
|
59 |
total_sequence = (
|
60 |
+
prompt_text + text[len(tokenizer.decode(encoded_prompt[0], clean_up_tokenization_spaces=True)) :] + "/n"
|
61 |
)
|
62 |
|
63 |
generated_sequences.append(total_sequence)
|