import gradio as gr
from transformers import pipeline

model = pipeline("text2text-generation", model="hafsa101010/mbart-neutralization")

def neutralize_text(input_text):
    result = model(input_text, max_length=128, do_sample=False)
    return result[0]["generated_text"]

demo = gr.Interface(
    fn=neutralize_text,
    inputs="text",
    outputs="text",
    title="Neutralizador para Lenguaje Inclusivo",
    description="Convierte frases en lenguaje no inclusivo s inclusivo."
)

demo.launch()