Text Classification
Transformers
PyTorch
Safetensors
English
bert
prompt-classification
text-embeddings-inference
# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("jeanvydes/jane")
model = AutoModelForSequenceClassification.from_pretrained("jeanvydes/jane")Quick Links
Prompt Classification
Finedtuned bert-uncased-model for prompt classification.
Uses
Classify Prompts into Tasks/Categories for select an appropiate LLM for it.
How to Get Started with the Model
Use the code below to get started with the model.
Training Details
Training Data
Fined tuned on https://huggingface.co/datasets/jeanvydes/llm-routing-text-classification
Model Card Authors [optional]
Contact
- Downloads last month
- 7
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="jeanvydes/jane")