nijatzeynalov's picture
Upload README.md with huggingface_hub
ec08f94 verified
|
raw
history blame contribute delete
No virus
981 Bytes
metadata
{}

IT Support Ticket Classification

Description: Automatically categorize and prioritize IT support tickets based on their text descriptions, enabling more efficient resolution and customer support.

How to Use

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

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