interneuronai/

bank_melli_iran_customer_service_chatbot_bart

    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))