|
import torch |
|
from transformers import pipeline |
|
import gradio as gr |
|
|
|
|
|
model_id = "unsloth/Llama-3.2-1B" |
|
|
|
|
|
pipe = pipeline( |
|
"text-classification", |
|
model=model_id, |
|
torch_dtype=torch.bfloat16, |
|
device_map="auto" |
|
) |
|
|
|
|
|
pipe.model.config.id2label = {0: 'saudação', 1: 'fim de conversa', 2: 'outro'} |
|
pipe.model.config.label2id = {'saudação': 0, 'fim de conversa': 1, 'outro': 2} |
|
|
|
|
|
def classify_text(text): |
|
result = pipe(text) |
|
return result[0]['label'] + ": " + str(result[0]['score']) |
|
|
|
|
|
iface = gr.Interface( |
|
fn=classify_text, |
|
inputs=gr.Textbox(label="Digite o texto"), |
|
outputs=gr.Label(label="Classificação"), |
|
title="Classificação Textual", |
|
description="Classifique seu texto como 'saudação', 'fim de conversa', ou 'outro'." |
|
) |
|
|
|
|
|
iface.launch() |