pszemraj commited on
Commit
7c0cbdc
1 Parent(s): 0230376

✨ add colab link

Browse files

Signed-off-by: peter szemraj <[email protected]>

Files changed (1) hide show
  1. app.py +9 -1
app.py CHANGED
@@ -489,7 +489,15 @@ if __name__ == "__main__":
489
  with demo:
490
  gr.Markdown("# Document Summarization with Long-Document Transformers")
491
  gr.Markdown(
492
- "An example use case for fine-tuned long document transformers. Model(s) are trained on [book summaries](https://hf.co/datasets/kmfoda/booksum). Architectures [in this demo](https://hf.co/spaces/pszemraj/document-summarization) are [LongT5-base](https://hf.co/pszemraj/long-t5-tglobal-base-16384-book-summary) and [Pegasus-X-Large](https://hf.co/pszemraj/pegasus-x-large-book-summary)."
 
 
 
 
 
 
 
 
493
  )
494
  with gr.Column():
495
  gr.Markdown("## Load Inputs & Select Parameters")
 
489
  with demo:
490
  gr.Markdown("# Document Summarization with Long-Document Transformers")
491
  gr.Markdown(
492
+ """An example use case for fine-tuned long document transformers. Model(s) are trained on [book summaries](https://hf.co/datasets/kmfoda/booksum). Architectures [in this demo](https://hf.co/spaces/pszemraj/document-summarization) are [LongT5-base](https://hf.co/pszemraj/long-t5-tglobal-base-16384-book-summary) and [Pegasus-X-Large](https://hf.co/pszemraj/pegasus-x-large-book-summary).
493
+
494
+ **Want more performance? Run this demo from a free Google Colab GPU:**.
495
+ <br>
496
+ <a href="https://colab.research.google.com/gist/pszemraj/52f67cf7326e780155812a6a1f9bb724/document-summarization-on-gpu.ipynb">
497
+ <img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab"/>
498
+ </a>
499
+ <br>
500
+ """
501
  )
502
  with gr.Column():
503
  gr.Markdown("## Load Inputs & Select Parameters")