Update README.md
Browse files
README.md
CHANGED
@@ -56,9 +56,7 @@ tokenizer = AutoTokenizer.from_pretrained('prajjwal1/bert-tiny')
|
|
56 |
|
57 |
pipe = pipeline('text-classification', model='shhossain/bert-tiny-book-text-classifier')
|
58 |
|
59 |
-
book_text = """
|
60 |
-
|
61 |
-
“Where’s the truck?” she asked, raising her voice slightly so that Ultron could hear her over the wind. It was a little demeaning to be cradled in the arms of one of his units like a child but it made flying possible, and that was a lot faster than any of the other possible modes of transportation."""
|
62 |
|
63 |
pipe(book_text) # LABEL_1
|
64 |
>> [{'label': 'LABEL_1', 'score': 0.9998537302017212}]
|
|
|
56 |
|
57 |
pipe = pipeline('text-classification', model='shhossain/bert-tiny-book-text-classifier')
|
58 |
|
59 |
+
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."""
|
|
|
|
|
60 |
|
61 |
pipe(book_text) # LABEL_1
|
62 |
>> [{'label': 'LABEL_1', 'score': 0.9998537302017212}]
|