nijatzeynalov's picture
Upload README.md with huggingface_hub
d45b882 verified
|
raw
history blame contribute delete
No virus
938 Bytes
metadata
{}

Customer Feedback Analysis

Description: Classify customer feedback based on sentiment and topic to identify improvement areas and strengthen customer engagement.

How to Use

Here is how to use this model to classify text into different categories:

    from transformers import AutoModelForSequenceClassification, AutoTokenizer
    
    model_name = "interneuronai/customer_feedback_analysis_bert"
    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))