Downloads · 30 days
12
9% of all-time downloads
branch-switch-v2
hitty28/branch-switch-v2
branch-switch-v2 is a text classification model from hitty28. Use it when you need a label for a piece of text. The card lists the license as apache-2.0.
This model classifies whether a user statement indicates a desire to switch branches or not.
All-time downloads
137
Public
Parameters
67M
268 MB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors268 MB · 100%
At a glance
- Task
- Text Classification
- License
- apache-2.0
- Model type
- distilbert
- Access
- Public
- Created
- Jun 18, 2025
- Updated
- Jun 18, 2025
- SHA
- cf71ca6e
Try a prompt
Base models
Datasets
Model card
From the Hugging Face model README
Branch Switch Classifier
This model classifies whether a user statement indicates a desire to switch branches or not.
Model Details
- Base Model: DistilBERT
- Task: Binary Text Classification
- Labels: True, False
Usage
from transformers import pipeline
classifier = pipeline("text-classification", model="hitty28/branch-switch-v2")
result = classifier("I want to switch to Mumbai branch")
print(result)
Training Data
Trained on custom dataset with statements about branch switching intentions.
- Task
- Text Classification
- Type
- distilbert
- License
- apache-2.0
- Languages
- en
- Created
- Jun 18, 2025
- Updated
- Jun 18, 2025