Spaces:
Paused
Paused
| import gradio as gr | |
| from transformers import AutoModelForCausalLM, AutoTokenizer | |
| import torch | |
| # Cargar el modelo y el tokenizador solo una vez para evitar recargas | |
| model_name = "BSC-LT/salamandra-2b" | |
| if "tokenizer" not in globals(): | |
| tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True) | |
| tokenizer.pad_token = tokenizer.eos_token # 🔹 Evita errores de atención | |
| if "model" not in globals(): | |
| model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16) | |
| model.eval() # 🔹 Optimiza la inferencia para que sea más rápida | |
| # Función de humanización de texto con enfoque en contenido persuasivo | |
| def humanize_text(input_text): | |
| system_prompt = ( | |
| "Reescribe el siguiente texto de manera más natural, clara y persuasiva, " | |
| "haciéndolo más cercano y atractivo para el lector. Reformula las oraciones " | |
| "para que sean más dinámicas y menos rígidas, pero sin perder información clave. " | |
| "Evita frases genéricas y reemplázalas por expresiones más auténticas. " | |
| "Aquí tienes un ejemplo de cómo debería sonar la reescritura:\n\n" | |
| "Ejemplo:\n" | |
| "Texto original: 'Nuestro software es la mejor opción para cualquier empresa.'\n" | |
| "Texto humanizado: 'Si buscas una herramienta que realmente optimice tu negocio, nuestro software puede ser justo lo que necesitas.'\n\n" | |
| "Ahora reescribe el siguiente texto:" | |
| ) | |
| prompt = f"{system_prompt}\n\nTexto original: {input_text}\n\nTexto humanizado:" | |
| inputs = tokenizer(prompt, return_tensors="pt", padding=True, truncation=True) | |
| outputs = model.generate( | |
| inputs.input_ids, | |
| attention_mask=inputs.attention_mask, | |
| max_new_tokens=140, # 🔹 Mantiene equilibrio entre reformulación y velocidad | |
| min_length=50, # 🔹 Evita respuestas demasiado cortas | |
| do_sample=True, # 🔹 Reactivamos la variabilidad para que reformule mejor | |
| temperature=0.78, # 🔹 Ajustamos la creatividad sin perder precisión | |
| top_p=0.9, # 🔹 Mantiene coherencia en la reescritura | |
| repetition_penalty=1.05, # 🔹 Evita repeticiones sin restringir demasiado | |
| num_return_sequences=1, # 🔹 Genera solo una respuesta bien formulada | |
| ) | |
| return tokenizer.decode(outputs[0], skip_special_tokens=True) | |
| # Interfaz en Gradio | |
| with gr.Blocks() as demo: | |
| gr.Markdown("# ✍️ Humanización de Texto con ALIA (Corrección de Reescritura)") | |
| 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") | |
| submit_button.click(humanize_text, inputs=input_text, outputs=output_text) | |
| demo.launch() | |