JoakimL13's picture
Update README.md
0f7463d verified
|
raw
history blame
493 Bytes
metadata
language:
  - en
pipeline_tag: question-answering
tags:
  - music
  - lyrics
  - cursing
datasets:
  - 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")

from datasets import load_dataset

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