Spaces:
Build error
Build error
bankholdup
commited on
Commit
•
5306456
1
Parent(s):
03808e1
Update app.py
Browse files
app.py
CHANGED
@@ -48,7 +48,7 @@ if st.button("Поехали", help="Может занять какое-то в
|
|
48 |
generated_sequence = generated_sequence.tolist()
|
49 |
text = tokenizer.decode(generated_sequence, clean_up_tokenization_spaces=True)
|
50 |
|
51 |
-
|
52 |
|
53 |
total_sequence = (
|
54 |
prompt_text + text[len(tokenizer.decode(encoded_prompt[0], clean_up_tokenization_spaces=True)) :]
|
|
|
48 |
generated_sequence = generated_sequence.tolist()
|
49 |
text = tokenizer.decode(generated_sequence, clean_up_tokenization_spaces=True)
|
50 |
|
51 |
+
text = text[: text.find("/n") if "/n" else None]
|
52 |
|
53 |
total_sequence = (
|
54 |
prompt_text + text[len(tokenizer.decode(encoded_prompt[0], clean_up_tokenization_spaces=True)) :]
|