voacado's picture
Update README.md
ed7aa49
metadata
license: mit
language:
  - en
pipeline_tag: text2text-generation
tags:
  - art

Dataset: https://www.kaggle.com/datasets/voanthony/two-sentence-horror-jan-2015-apr-2023

Space: https://huggingface.co/spaces/voacado/two-sentence-horror-bart

Model was fine-tuned from 'https://huggingface.co/facebook/bart-base' on Kaggle's P100 GPU (3 epochs took about 5 hours).

Output is uncased, can use 'https://huggingface.co/felflare/bert-restore-punctuation' to restore punctuation.