File size: 798 Bytes
7d0343b f49c538 0455597 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 |
---
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)
|