Update app.py

app.py

@@ -19,9 +19,9 @@ def infer(input_ids):
     output_sequences = model.generate(
         input_ids=input_ids,
         max_length=40,
-#        do_sample=False,
+        do_sample=False,
         num_return_sequences=1,
-        #num_beams=8,
+        num_beams=8,
         length_penalty=1.0,
         no_repeat_ngram_size=3,
       #  do_sample=True, top_p=0.9, temperature=0.9
@@ -44,7 +44,7 @@ if custom_sent == default_value:
 
 st.button('Сделать нетоксичным')
 
-encoded_prompt = tokenizer.encode(custom_sent, add_special_tokens=False, return_tensors="pt")
+encoded_prompt = tokenizer.encode(custom_sent, add_special_tokens=True, return_tensors="pt")
 if encoded_prompt.size()[-1] == 0:
     input_ids = None
 else: