robzchhangte commited on
Commit
76cabf8
·
verified ·
1 Parent(s): bbd32a5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,7 +1,7 @@
1
  from transformers import pipeline
2
 
3
  # Initialize the fill-mask pipeline with your chosen model
4
- unmasker = pipeline("fill-mask", model="10-bert-uncased-datav4")
5
 
6
  def predict(text):
7
  """
 
1
  from transformers import pipeline
2
 
3
  # Initialize the fill-mask pipeline with your chosen model
4
+ unmasker = pipeline("fill-mask", model="bert-base-uncased")
5
 
6
  def predict(text):
7
  """