--- license: mit datasets: - imdb - rotten_tomatoes language: - en metrics: - accuracy pipeline_tag: text-classification --- # Book Test Classifier Classify book text ## Model Details ### Model Description This model is finetuned on bert-tiny for classifying book text. - **Developed by:** [shhossain](https://github.com/shhossain) - **Model type:** [Bert] - **Language(s) (NLP):** [English] - **License:** [MIT] - **Finetuned from model [Bert-Tiny]:** [bert-tiny](https://huggingface.co/prajjwal1/bert-tiny) ## Uses ```python from transformers import pipeline, AutoTokenizer tokenizer = AutoTokenizer.from_pretrained('prajjwal1/bert-tiny') pipe = pipeline('text-classification', 'shhossain/bert-tiny-book-text-classifier') ```