import gradio as gr
from transformers import pipeline

# Carregar o modelo
pipe = pipeline(model="MoritzLaurer/mDeBERTa-v3-base-xnli-multilingual-nli-2mil7")

# Função de inferência para o Gradio
def classify_text(text):
    result = pipe(text, candidate_labels=['saudação', 'fim de conversa', 'outro'])
    return result

# Criar a interface do Gradio
interface = gr.Interface(fn=classify_text, 
                         inputs="text", 
                         outputs="json", 
                         title="Classificador de Textos - ZEROSHOT", 
                         description="Classifique um texto em uma das categorias: saudação, fim de conversa ou outro.")

# Rodar a interface
interface.launch()