| from transformers import pipeline | |
| # google bert tasks | |
| # fill-mask | |
| # feature extraction | |
| # have to use more specific models/other models for other tasks | |
| # for mask gen | |
| unmasker = pipeline('fill-mask', model='bert-base-uncased') | |
| results = unmasker("Obama is [MASK].") | |
| for prediction in results: | |
| print(f"Score: {prediction['score']:.4f} | Prediction: {prediction['sequence']}") |