| {} | |
| ### Bank Melli Iran Customer Service Chatbot | |
| **Description:** Classify customer inquiries into predefined categories and provide automated responses, improving customer service and reducing response time | |
| ## How to Use | |
| Here is how to use this model to classify text into different categories: | |
| from transformers import AutoModelForSequenceClassification, AutoTokenizer | |
| model_name = "interneuronai/bank_melli_iran_customer_service_chatbot_bart" | |
| 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)) |