callcenter-ticket-classifier / inference_example.py
Kahouli's picture
Initial model upload
a89ad88 verified
# Exemple d'inférence simple
from transformers import pipeline
# Charger le pipeline
classifier = pipeline("text-classification", model="./")
# Classifier un ticket
text = "Mon imprimante ne fonctionne plus"
result = classifier(text)
print(f"Catégorie: {result[0]['label']}")
print(f"Confiance: {result[0]['score']:.2%}")