File size: 744 Bytes
be939ac
 
 
 
 
 
0f7463d
 
bb4472e
 
4e89252
be939ac
fd0f955
 
 
 
 
 
635e95a
ffde4de
 
 
 
 
 
635e95a
 
bb4472e
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
---
language:
- en
pipeline_tag: question-answering
tags:
- music
- lyrics
- cursing
datasets:
- sebastiandizon/genius-song-lyrics
library_name: allennlp
---

# Load model directly
from transformers import AutoTokenizer, AutoModelForDocumentQuestionAnswering

tokenizer = AutoTokenizer.from_pretrained("impira/layoutlm-document-qa")
model = AutoModelForDocumentQuestionAnswering.from_pretrained("impira/layoutlm-document-qa")

# Load model directly
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM

tokenizer = AutoTokenizer.from_pretrained("facebook/bart-large-cnn")
model = AutoModelForSeq2SeqLM.from_pretrained("facebook/bart-large-cnn")

from datasets import load_dataset

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