File size: 981 Bytes
ec08f94
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
---
{}
---
### 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))