Update app.py

app.py

@@ -13,13 +13,12 @@ if "model" not in globals():
     model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16)
     model.eval()  
 
-# 🔹 Función de humanización con el prompt más efectivo
+# 🔹 Función de humanización optimizada
 def humanize_text(input_text):
     system_prompt = (
-        "Reescribe el siguiente texto de manera más clara, natural y atractiva, "
-        "sin cambiar su significado. Reformula frases rígidas y estructuradas para "
-        "que sean más fluidas y conversacionales, pero sin perder precisión. "
-        "Evita tecnicismos y burocracia innecesaria."
+        "Reescribe el siguiente texto con un tono más natural y fluido, "
+        "sin cambiar su significado ni agregar información nueva. "
+        "Evita repeticiones y burocracia innecesaria, asegurando que suene humano y claro."
     )
 
     prompt = f"{system_prompt}\n\nTexto original: {input_text}\n\nTexto humanizado:"
@@ -29,12 +28,12 @@ def humanize_text(input_text):
     outputs = model.generate(
         inputs.input_ids,
         attention_mask=inputs.attention_mask,
-        max_new_tokens=130,  # 🔹 Equilibrio entre reformulación y velocidad
+        max_new_tokens=120,  # 🔹 Equilibrio entre reformulación y velocidad
         min_length=50,  # 🔹 Evita respuestas demasiado cortas
         do_sample=True,  # 🔹 Mantiene variabilidad sin ralentizar
-        temperature=0.75,  # 🔹 Buen balance entre creatividad y rapidez
-        top_p=0.9,  # 🔹 Mantiene coherencia en la reescritura
-        repetition_penalty=1.05,  # 🔹 Reduce repeticiones sin afectar fluidez
+        temperature=0.8,  # 🔹 Aumentamos un poco la creatividad
+        top_p=0.85,  # 🔹 Más variabilidad sin afectar precisión
+        repetition_penalty=1.02,  # 🔹 Evita repeticiones sin ser tan restrictivo
         num_return_sequences=1,  # 🔹 Solo una respuesta bien formulada
     )
 
@@ -42,7 +41,7 @@ def humanize_text(input_text):
 
 # Interfaz en Gradio
 with gr.Blocks() as demo:
-    gr.Markdown("# ✍️ Humanización de Texto con ALIA&SOLID")
+    gr.Markdown("# ✍️ Humanización de Texto con ALIA")
     input_text = gr.Textbox(label="Pega aquí el texto generado por IA para humanizar")
     output_text = gr.Textbox(label="Texto humanizado por ALIA", interactive=False)
     submit_button = gr.Button("Humanizar Texto")