Ahmadzei's picture
added 3 more tables for large emb model
5fa1a76
``
</pt>
<tf>
Tokenize the text and return theinput_ids` as TensorFlow tensors:
from transformers import AutoTokenizer
tokenizer = AutoTokenizer.from_pretrained("stevhliu/my_awesome_billsum_model")
inputs = tokenizer(text, return_tensors="tf").input_ids
Use the [~transformers.generation_tf_utils.TFGenerationMixin.generate] method to create the summarization.