Update app.py

app.py

@@ -20,7 +20,7 @@ def infer(input_ids):
 #        do_sample=False,
         num_return_sequences=1,
         num_beams=8,
-        length_penalty=1.0,
+        length_penalty=1.5,
         no_repeat_ngram_size=4,
        # do_sample=True, top_p=0.9, temperature=0.9
     )
@@ -32,7 +32,7 @@ default_value = "Еще один петух решил легко заработ
 st.title("Демо детоксификации на ruT5")
 sent = st.text_area("Text", default_value)
 
-
+st.button('Сделать нетактичным')
 
 encoded_prompt = tokenizer.encode(sent, add_special_tokens=False, return_tensors="pt")
 if encoded_prompt.size()[-1] == 0: