companyx_customer_support_ticket_routing_distilbert

CompanyX Customer Support Ticket Routing

Description: Automatically route customer support tickets to relevant teams based on issue descriptions, speeding up resolution time and enhancing customer experience.

How to Use

Here is how to use this model to classify text into different categories:

    from transformers import AutoModelForSequenceClassification, AutoTokenizer
    
    model_name = "interneuronai/companyx_customer_support_ticket_routing_distilbert"
    model = AutoModelForSequenceClassification.from_pretrained(model_name)
    tokenizer = AutoTokenizer.from_pretrained(model_name)
    
    def classify_text(text):
        inputs = tokenizer(text, return_tensors="pt", padding=True, truncation=True, max_length=512)
        outputs = model(**inputs)
        predictions = outputs.logits.argmax(-1)
        return predictions.item()
    
    text = "Your text here"
    print("Category:", classify_text(text))