cci-capstone / README.md
raghavdw's picture
Update README.md
838fc78 verified
metadata
tags:
  - text-classification
library_name: transformers
language: en
license: apache-2.0

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)