Shushant Pudasaini commited on
Commit
698130b
1 Parent(s): 8bd1280

updated model card

Browse files

Usage model card update

Files changed (1) hide show
  1. README.md +20 -1
README.md CHANGED
@@ -1 +1,20 @@
1
- # 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.
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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}."))