mirror of
https://github.com/huggingface/transformers.git
synced 2025-08-01 18:51:14 +06:00
19 lines
569 B
Python
19 lines
569 B
Python
"""
|
|
Show how to use HuggingFace's PyTorch implementation of Google's BERT Model.
|
|
"""
|
|
from .bert_model import BERT
|
|
from .prepare_inputs import DataPreprocessor
|
|
|
|
bert_model = BERT()
|
|
bert_model.load_from('.')
|
|
data_processor = DataProcessor(encoder_file_path='.')
|
|
|
|
input_sentence = "We are playing with the BERT model."
|
|
print("BERT inputs: {}".format(input_sentence))
|
|
|
|
tensor_input = data_processor.encode(input_sentence)
|
|
tensor_output = bert_model(prepared_input)
|
|
output_sentence = data_processor.decode(tensor_output)
|
|
|
|
print("BERT predicted: {}".format(output_sentence))
|