| 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() |