shhossain commited on
Commit
b4f47de
1 Parent(s): ac562f3

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -3
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 = """Skye tried very hard not to look at the shattered glass that was all over the city’s paths and roads, and pretty much everywhere, really. It wasn’t quite as much damage as the Hulk had apparently done under Wanda’s influence but it was pretty bad nonetheless. Pretty much the only comfort she had, small as it was, was that she probably hadn’t killed or seriously injured anyone since an earthquake and an Avengers fight at once appeared to have done wonders for keeping the streets clear.
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}]