JoakimL13's picture
Update README.md
fd0f955 verified
|
raw
history blame
429 Bytes
---
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")