nijatzeynalov commited on
Commit
bbeac77
1 Parent(s): 6f64a1b

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
+ ### Bank Melli Iran Customer Service Chatbot
5
+
6
+ **Description:** Classify customer inquiries into predefined categories and provide automated responses, improving customer service and reducing response time
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/bank_melli_iran_customer_service_chatbot_bart"
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))