boltuix commited on
Commit
9f3cb89
·
verified ·
1 Parent(s): 1200629

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -90,8 +90,8 @@ from transformers import pipeline
90
  # 📢 Start demo
91
  print("\n🔤 Masked Language Model (MLM) Demo")
92
 
93
- # 🧠 Load masked language model
94
- mlm_pipeline = pipeline("fill-mask", model="bert-base-uncased")
95
 
96
  # ✍️ Masked sentences
97
  masked_sentences = [
 
90
  # 📢 Start demo
91
  print("\n🔤 Masked Language Model (MLM) Demo")
92
 
93
+ # 🧠 Load masked language model : eg boltuix/bert-lite
94
+ mlm_pipeline = pipeline("fill-mask", model="boltuix/bert-lite")
95
 
96
  # ✍️ Masked sentences
97
  masked_sentences = [