Update app.py
Browse files
app.py
CHANGED
|
@@ -19,7 +19,7 @@ def infer(input_ids):
|
|
| 19 |
max_length=40,
|
| 20 |
# do_sample=False,
|
| 21 |
num_return_sequences=1,
|
| 22 |
-
num_beams=8,
|
| 23 |
length_penalty=1.0,
|
| 24 |
#no_repeat_ngram_size=4,
|
| 25 |
do_sample=True, top_p=0.9, temperature=0.7
|
|
@@ -27,14 +27,22 @@ def infer(input_ids):
|
|
| 27 |
|
| 28 |
return output_sequences
|
| 29 |
default_value = "всегда ненавидел этих ублюдочных тварей"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 30 |
|
| 31 |
#prompts
|
| 32 |
st.title("Демо детоксификации на ruT5")
|
| 33 |
-
sent = st.
|
|
|
|
|
|
|
|
|
|
| 34 |
|
| 35 |
st.button('Сделать нетоксичным')
|
| 36 |
|
| 37 |
-
encoded_prompt = tokenizer.encode(
|
| 38 |
if encoded_prompt.size()[-1] == 0:
|
| 39 |
input_ids = None
|
| 40 |
else:
|
|
|
|
| 19 |
max_length=40,
|
| 20 |
# do_sample=False,
|
| 21 |
num_return_sequences=1,
|
| 22 |
+
#num_beams=8,
|
| 23 |
length_penalty=1.0,
|
| 24 |
#no_repeat_ngram_size=4,
|
| 25 |
do_sample=True, top_p=0.9, temperature=0.7
|
|
|
|
| 27 |
|
| 28 |
return output_sequences
|
| 29 |
default_value = "всегда ненавидел этих ублюдочных тварей"
|
| 30 |
+
examples = [default_value,
|
| 31 |
+
"убила бы этих выродков и их родителей.",
|
| 32 |
+
"пошли вы сука все на хуй со своим коронавирусом...",
|
| 33 |
+
"Перед барином выслуживаешься холоп?",
|
| 34 |
+
"просто , зарплату шакалов отрабатывают , а больше на крысенышей похожи..."]
|
| 35 |
|
| 36 |
#prompts
|
| 37 |
st.title("Демо детоксификации на ruT5")
|
| 38 |
+
sent = st.selectbox("Пример", examples)
|
| 39 |
+
custom_sent = st.text_area("Исходный текст", default_value)
|
| 40 |
+
if custom_sent == default_value:
|
| 41 |
+
custom_sent = sent
|
| 42 |
|
| 43 |
st.button('Сделать нетоксичным')
|
| 44 |
|
| 45 |
+
encoded_prompt = tokenizer.encode(custom_sent, add_special_tokens=False, return_tensors="pt")
|
| 46 |
if encoded_prompt.size()[-1] == 0:
|
| 47 |
input_ids = None
|
| 48 |
else:
|