app.py

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

title = "Spanish Text Simplification"
description = "Simplificación de noticias en español"
examples = [
    ["""
  El lanzamiento corresponde al alcalde de Vitoria-Gasteiz, aunque delega en los grupos políticos por rotación el lanzamiento del cohete. El turno se delega a cada grupo, de más a menos representación. Una tradición que implantó en 1999 Alfonso Alonso, ya que José Ángel Cuerda prendió año tras año el cohete durante sus 20 años como alcalde. Antes, en el Franquismo, los sucesivos alcaldes lanzaron el cohete, aunque también en otras ocasiones lo fue el responsable de la comisión de festejos.
"""],
    ["""
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
"""]
]

# Cargar los modelos desde Hugging Face
demo = gr.Interface.load("crrodrvi/modelo_simplificacion_bart",
                         theme='huggingface', 
                         inputs="textbox", 
                         # outputs="textbox",
                         title = title,
                         description=description, 
                         examples=examples)

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