Update app.py
Browse files
app.py
CHANGED
@@ -46,7 +46,7 @@ if sent == "--- свой текст ---":
|
|
46 |
|
47 |
st.button('Сделать нетоксичным')
|
48 |
|
49 |
-
encoded_prompt = tokenizer.encode(
|
50 |
if encoded_prompt.size()[-1] == 0:
|
51 |
input_ids = None
|
52 |
else:
|
|
|
46 |
|
47 |
st.button('Сделать нетоксичным')
|
48 |
|
49 |
+
encoded_prompt = tokenizer.encode(sent, add_special_tokens=True, return_tensors="pt")
|
50 |
if encoded_prompt.size()[-1] == 0:
|
51 |
input_ids = None
|
52 |
else:
|