Update app.py

app.py

@@ -16,7 +16,7 @@ def infer(input_ids):
     
     output_sequences = model.generate(
         input_ids=input_ids,
-        max_length=60,
+        max_length=40,
         do_sample=False,
         num_return_sequences=1,
         num_beams=32,
@@ -29,7 +29,7 @@ default_value = "А ну иди сюда, придурок"
 
 #prompts
 st.title("Дело детоксификации на ruT5")
-sent = st.text_area("Text", default_value, height = 275)
+sent = st.text_area("Text", default_value, height = 27)
 
 
 
@@ -56,7 +56,7 @@ for generated_sequence_idx, generated_sequence in enumerate(output_sequences):
 
     # Add the prompt at the beginning of the sequence. Remove the excess text that was used for pre-processing
     total_sequence = (
-        sent + text[len(tokenizer.decode(encoded_prompt[0], clean_up_tokenization_spaces=True)) :]
+        text
     )
 
     generated_sequences.append(total_sequence)