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