Skip to content

AzimAhmedBijapur/Grievance-Severity-Classification-Distilbert

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 

Repository files navigation

Fine-tuned distilbert for classifying complaints by severity (low, medium, high)

Deployment link

https://huggingface.co/azim17/Gbert-85

Use a pipeline as a high-level helper

from transformers import pipeline
pipe = pipeline("text-classification", model="azim17/Gbert-85")

Load model directly

from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("azim17/Gbert-85")
model = AutoModelForSequenceClassification.from_pretrained("azim17/Gbert-85")

About

Fine-tuned a DistilBERT model to classify complaints based on severity.

Resources

Stars

Watchers

Forks