app.py

import gradio as gr
from transformers import pipeline

# Cargar el modelo pre-entrenado
gender_classifier = pipeline('text-classification', model='Dannel/Gender_Classifier')

def infer_gender(name):
    """
    Infiere el género de una persona a partir de su nombre.
    
    Args:
        name (str): El nombre de la persona.
        
    Returns:
        str: El género predicho ('Male' o 'Female').
    """
    # Hacer la predicción utilizando el modelo cargado
    prediction = gender_classifier([name])[0]
    
    return prediction['label']

# Crear la interfaz de Gradio
demo = gr.Interface(
    fn=infer_gender,
    inputs=gr.Textbox(label="Nombre"),
    outputs=gr.Label(label="Género predicho"),
    title="Clasificador de Género",
    description="Ingresa un nombre para predecir su género."
)

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