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