nijatzeynalov's picture
Upload README.md with huggingface_hub
975fb6f verified
|
raw
history blame
985 Bytes
metadata
{}

Customer Support Ticket Classification

Description: Categorize customer support tickets based on their content to improve the efficiency of the support team and provide faster resolution times.

How to Use

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

    from transformers import AutoModelForSequenceClassification, AutoTokenizer
    
    model_name = "interneuronai/customer_support_ticket_classification_pegasus"
    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))