nijatzeynalov commited on
Commit
975fb6f
1 Parent(s): 3e8c02d

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +24 -0
README.md ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ {}
3
+ ---
4
+ ### Customer Support Ticket Classification
5
+
6
+ **Description:** Categorize customer support tickets based on their content to improve the efficiency of the support team and provide faster resolution times.
7
+
8
+ ## How to Use
9
+ Here is how to use this model to classify text into different categories:
10
+
11
+ from transformers import AutoModelForSequenceClassification, AutoTokenizer
12
+
13
+ model_name = "interneuronai/customer_support_ticket_classification_pegasus"
14
+ model = AutoModelForSequenceClassification.from_pretrained(model_name)
15
+ tokenizer = AutoTokenizer.from_pretrained(model_name)
16
+
17
+ def classify_text(text):
18
+ inputs = tokenizer(text, return_tensors="pt", padding=True, truncation=True, max_length=512)
19
+ outputs = model(**inputs)
20
+ predictions = outputs.logits.argmax(-1)
21
+ return predictions.item()
22
+
23
+ text = "Your text here"
24
+ print("Category:", classify_text(text))