hafsa101010's picture
Create app.py
d3c69ce verified
raw
history blame contribute delete
501 Bytes
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()