raghavdw/cci-capstone

Airline Customer Service Intelligence Model

Last Updated: 2025-01-07 22:14:37 Using Checkpoint: checkpoint-1170

Model Configuration

• Base: RoBERTa
• Multi-task Classification (Intent, Sentiment, Topic)
• Checkpoint: checkpoint-1170

Usage

from transformers import AutoModelForSequenceClassification, AutoTokenizer

model = AutoModelForSequenceClassification.from_pretrained("raghavdw/cci-capstone")
tokenizer = AutoTokenizer.from_pretrained("raghavdw/cci-capstone")

text = "I need help with my flight booking"
inputs = tokenizer(text, return_tensors="pt")
outputs = model(**inputs)