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)
- Downloads last month
- 1
Inference Providers
NEW
This model is not currently available via any of the supported third-party Inference Providers, and
the model is not deployed on the HF Inference API.