Update README.md

README.md

@@ -8,6 +8,10 @@ tags:
 ---
 
 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'. Output is uncased, can use 'https://huggingface.co/felflare/bert-restore-punctuation' to restore punctuation.
+
+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.