Downloads · 30 days
64
6% of all-time downloads
shhossain/bert-tiny-book-text-classifier
bert-tiny-book-text-classifier is a text classification model from shhossain. 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.
Classify book text (mostly fictional book)
Downloads · 30 days
64
6% of all-time downloads
All-time downloads
1K
Public
Parameters
4.4M
70.2 MB on disk
Likes
3
Public
Click a slice to open those files.
.bin17.6 MB · 49%
From the Hugging Face model README
Classify book text (mostly fictional book)
This model is finetuned on bert-tiny for classifying book text.
from transformers import pipeline, AutoTokenizer
tokenizer = AutoTokenizer.from_pretrained('prajjwal1/bert-tiny')
pipe = pipeline('text-classification', model='shhossain/bert-tiny-book-text-classifier')
book_text = """Shen Yuanye didn’t expect to go back to Huang Ni’s matter again, this matter of being killed without finding the murderer, who else could be beside her."""
pipe(book_text) # LABEL_1
>> [{'label': 'LABEL_1', 'score': 0.9998537302017212}]
normal_text = """I am so sorry this is a day late, guys. Unfortunately, my internet was down so it was out of my control. Its still intermittent but hopefully it will be fine by next week. Hopefully the fact that Skye and Pietro are back in form will help make up for it."""
pipe(normal_text) # LABEL_0
>> [{'label': 'LABEL_0', 'score': 0.9984021782875061}]