Downloads · 30 days
6
20% of all-time downloads
Bookmus
methodya/Bookmus
Bookmus is a text classification model from methodya. 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 model predicts book genres based on their titles. It's trained on Arabic and English book titles.
All-time downloads
30
Public
Repo size
541 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.h5541 MB · 100%
At a glance
- Task
- Text Classification
- Library
- transformers
- License
- mit
- Model type
- bert
- Access
- Public
- Created
- Nov 19, 2023
- Updated
- Nov 19, 2023
- SHA
- 04555974
Try a prompt
Model card
From the Hugging Face model README
Book Genre Prediction Model
This model predicts book genres based on their titles. It's trained on Arabic and English book titles.
Overview
The model is built using a BERT-based architecture and is trained to classify book titles into various genres. It leverages tokenization specific to Arabic and English languages for effective processing.
How to Use
Installation
To use this model, first, install the transformers library:
pip install transformers
### Usage
from transformers import AutoTokenizer, TFAutoModelForSequenceClassification
model_name = "Book Genre Prediction Model"
# Load the tokenizer and model
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = TFAutoModelForSequenceClassification.from_pretrained(model_name)
- Task
- Text Classification
- Library
- transformers
- Type
- bert
- License
- mit
- Created
- Nov 19, 2023
- Updated
- Nov 19, 2023