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)
Downloads last month
342
Safetensors
Model size
125M params
Tensor type
F32
·
Inference Examples
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.