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") | |