app.py

import gradio as gr
import torch
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM

model_dir = "crrodrvi/modelo_simplificacion_bart"

# Cargar tokenizer y modelo desde Hugging Face
tokenizer = AutoTokenizer.from_pretrained(model_dir)
model = AutoModelForSeq2SeqLM.from_pretrained(model_dir)

# Función de inferencia para simplificación de texto
def inference(input_text):
    inputs = tokenizer(input_text, return_tensors="pt", max_length=512, truncation=True, padding="max_length")
    
    # Generar la secuencia de salida del modelo
    with torch.no_grad():
        outputs = model.generate(**inputs)
    
    # Decodificar la secuencia generada en texto legible
    output_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
    
    return output_text
    
title = "Spanish Text Simplification"
description = "Simplificación de noticias en español"
examples = [
    ["""
    El primer ministro británico, Rishi Sunak, ha defendido ante la Cámara de los Comunes el papel de Reino Unido en el derribo de drones durante el ataque de Irán contra Israel en represalia por el bombardeo contra el Consulado iraní en la capital de Siria, Damasco. "En apoyo de la propia acción defensiva de Israel, Reino Unido se unió a un esfuerzo internacional liderado por Estados Unidos, junto con Francia y sus socios en la región, que interceptó casi todos los misiles, salvando vidas en Israel y sus vecinos", ha explicado. Sunak ha indicado que Irán "tiene la intención de sembrar el caos en su propio patio trasero para desestabilizar aún más Oriente Próximo" y ha precisado que "todas las partes deben mostrar moderación" para evitar una escalada
    """],
    ["""
    La semana comienza con un descenso de las temperaturas en la mitad norte el lunes y el martes, según la predicción del portavoz de la Agencia Estatal de Meteorología (AEMET), Rubén del Campo.  En concreto, este lunes la bajada será de hasta 10ºC en el Cantábrico con respecto a los valores del domingo. Además, los vientos del norte serán intensos en la costa gallega, Valle del Ebro y Ampurdán catalán durante el martes, jornada en la que el norte de Oviedo, Santander o Pamplona no llegarán a superar los 15ºC. Ya el miércoles, habrá una bajada de temperaturas más generalizada
    """]
]

# Crear la interfaz de Gradio
demo = gr.Interface(
    fn=inference,
    inputs=gr.Textbox(label="Input Text", placeholder="Enter text here...", lines=10),
    outputs=gr.Textbox(label="Output Simplified Text", placeholder="Simplified text will appear here..."),
    title=title,
    description=description,
    examples=examples
)

# Lanzar la aplicación
demo.launch(share=True)