|
|
--- |
|
|
license: mit |
|
|
--- |
|
|
|
|
|
# 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: |
|
|
|
|
|
```bash |
|
|
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) |
|
|
|
|
|
|
|
|
|
