JoakimL13's picture
Update README.md
fd0f955 verified
|
raw
history blame
429 Bytes
metadata
language:
  - en
pipeline_tag: question-answering
tags:
  - music

from datasets import load_dataset

ds = load_dataset("sebastiandizon/genius-song-lyrics")

Load model directly

from transformers import AutoTokenizer, AutoModelForDocumentQuestionAnswering

tokenizer = AutoTokenizer.from_pretrained("impira/layoutlm-document-qa") model = AutoModelForDocumentQuestionAnswering.from_pretrained("impira/layoutlm-document-qa")