JoakimL13's picture
Update README.md
ffde4de verified
|
raw
history blame contribute delete
No virus
744 Bytes
metadata
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")