File size: 429 Bytes
be939ac
 
 
 
 
 
 
 
 
fd0f955
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
---
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")