Shushant Pudasaini
commited on
Commit
•
698130b
1
Parent(s):
8bd1280
updated model card
Browse filesUsage model card update
README.md
CHANGED
@@ -1 +1,20 @@
|
|
1 |
-
#
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# NepNewsBERT
|
2 |
+
## Masked Language Model for nepali language trained on nepali news scrapped from different nepali news website. The data set contained about 10 million of nepali sentences mainly related to nepali news.
|
3 |
+
|
4 |
+
## Usage
|
5 |
+
|
6 |
+
from transformers import AutoTokenizer, AutoModelForMaskedLM
|
7 |
+
|
8 |
+
tokenizer = AutoTokenizer.from_pretrained("Shushant/NepNewsBERT")
|
9 |
+
|
10 |
+
model = AutoModelForMaskedLM.from_pretrained("Shushant/NepNewsBERT")
|
11 |
+
|
12 |
+
from transformers import pipeline
|
13 |
+
|
14 |
+
fill_mask = pipeline(
|
15 |
+
"fill-mask",
|
16 |
+
model=model,
|
17 |
+
tokenizer=tokenizer,
|
18 |
+
)
|
19 |
+
from pprint import pprint
|
20 |
+
pprint(fill_mask(f"तिमीलाई कस्तो {tokenizer.mask_token}."))
|