How to use pretrain model

#4
by Tizzzzy - opened

Hi,
How can I just play around with the pretrain model. For example, give the model a sentence and let it predict the output. What are the next steps for:

from transformers import AutoTokenizer, AutoModel

tokenizer = AutoTokenizer.from_pretrained("nlpaueb/legal-bert-base-uncased")
model = AutoModel.from_pretrained("nlpaueb/legal-bert-base-uncased")

Sign up or log in to comment