Downloads · 30 days
10
24% of all-time downloads
bert-job-chatbot
razaque/bert-job-chatbot
bert-job-chatbot is a text classification model from razaque. Use it when you need a label for a piece of text. It is set up for transformers. The card lists the license as mit.
This is a fine-tuned BERT model for job-related query classification. It was trained on a custom dataset for understanding and classifying job-related questions.
All-time downloads
41
Public
Parameters
109M
443 MB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors438 MB · 100%
At a glance
- Task
- Text Classification
- Library
- transformers
- License
- mit
- Model type
- bert
- Access
- Public
- Created
- Nov 26, 2024
- Updated
- Nov 27, 2024
- SHA
- 497fc947
Try a prompt
Base models
Model card
From the Hugging Face model README
BERT Job Chatbot
This is a fine-tuned BERT model for job-related query classification. It was trained on a custom dataset for understanding and classifying job-related questions.
Usage
from transformers import AutoModelForSequenceClassification, AutoTokenizer
model_name = "razaque/bert-job-chatbot"
model = AutoModelForSequenceClassification.from_pretrained(model_name)
tokenizer = AutoTokenizer.from_pretrained(model_name)
query = "What are the best software engineering roles?"
inputs = tokenizer(query, return_tensors="pt", truncation=True, padding="max_length", max_length=512)
outputs = model(**inputs)
predicted_class = outputs.logits.argmax(dim=-1).item()
print(f"Predicted class: {predicted_class}")
- Task
- Text Classification
- Library
- transformers
- Type
- bert
- License
- mit
- Languages
- en
- Created
- Nov 26, 2024
- Updated
- Nov 27, 2024