import gradio as gr
examples = [
    ['El autor se perfila, a los 50 años de su muerte, como uno de los grandes de su siglo', 'cultura, sociedad, economia, salud, deportes'],
    ['Elecciones 2023, estos son los candidatos a presidente', 'cultura, sociedad, economia, salud, deportes, política'],
    ['Lionel Messi finalmente jugará en un equipo de Miami', 'cultura, sociedad, economia, salud, deportes'],
    ['El dólar volvio a subir', 'cultura, sociedad, economia, salud, deportes']
]
def get_zero_shot_io():
    model_name = "Recognai/bert-base-spanish-wwm-cased-xnli"
    hf_summarization_io_api = gr.load(
        model_name,
        src="models",
        hf_token="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",
        examples=examples,
        title=f'Zeroshot {model_name}',
        description='La primera inferencia tarda un poco por que carga el modelo, luego es más rapido'
    )
    return hf_summarization_io_api

get_zero_shot_io().launch()