Update app.py

app.py

@@ -133,7 +133,7 @@ def humanize_text(text):
     input_ids = st.session_state.tokenizer(
         prompt,
         return_tensors="pt",
-        max_length=1024,
+        max_length=512,
         truncation=True
     ).input_ids
 
@@ -142,12 +142,12 @@ def humanize_text(text):
         input_ids,
         max_length=1024,
         min_length=len(text.split()),
-        do_sample=True,
+        do_sample=False,
         temperature=0.1,
         top_p=0.95,
-        num_beams=3,
-        repetition_penalty=1.2,
-        length_penalty=2.0
+        num_beams=2,
+        repetition_penalty=1.1,
+        length_penalty=1.0
     )
     result = st.session_state.tokenizer.decode(outputs[0], skip_special_tokens=True)
     result = clean_generated_text(result)