Skip to content

Instantly share code, notes, and snippets.

@HeenaR17

HeenaR17/fill-mask.py

Created December 21, 2020 16:53
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save HeenaR17/972018583dd119aa8093fbb451b476a1 to your computer and use it in GitHub Desktop.
Save HeenaR17/972018583dd119aa8093fbb451b476a1 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
fill-mask.py
from transformers import pipeline
def create_pipeline(model_name):
return pipeline('fill-mask', model=model_name)
predict_mask = create_pipeline("bert-base-uncased")
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment