shhossain commited on
Commit
46453c6
·
1 Parent(s): 5cbdec0

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +41 -0
README.md CHANGED
@@ -1,3 +1,44 @@
1
  ---
2
  license: mit
 
 
 
 
 
 
 
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: mit
3
+ datasets:
4
+ - imdb
5
+ - rotten_tomatoes
6
+ language:
7
+ - en
8
+ metrics:
9
+ - accuracy
10
+ pipeline_tag: text-classification
11
  ---
12
+ # Book Test Classifier
13
+
14
+ Classify book text
15
+
16
+
17
+ ## Model Details
18
+
19
+ ### Model Description
20
+
21
+ This model is finetuned on bert-tiny for classifying book text.
22
+
23
+
24
+
25
+ - **Developed by:** [shhossain](https://github.com/shhossain)
26
+ - **Model type:** [Bert]
27
+ - **Language(s) (NLP):** [English]
28
+ - **License:** [MIT]
29
+ - **Finetuned from model [Bert-Tiny]:** [bert-tiny](https://huggingface.co/prajjwal1/bert-tiny)
30
+
31
+
32
+
33
+ ## Uses
34
+
35
+
36
+ ```python
37
+ from transformers import pipeline, AutoTokenizer
38
+ tokenizer = AutoTokenizer.from_pretrained('prajjwal1/bert-tiny')
39
+
40
+ pipe = pipeline('text-classification', 'shhossain/bert-tiny-book-text-classifier')
41
+ ```
42
+
43
+
44
+