|
|
|
--- |
|
language: en |
|
tags: |
|
- sagemaker |
|
- bart |
|
- summarization |
|
license: apache-2.0 |
|
datasets: |
|
- samsum |
|
--- |
|
|
|
## bart-base-samsum |
|
This model was trained using Amazon SageMaker and the new Hugging Face Deep Learning container. |
|
|
|
For more information look at: |
|
- [π€ Transformers Documentation: Amazon SageMaker](https://huggingface.co/transformers/sagemaker.html) |
|
- [Example Notebooks](https://github.com/huggingface/notebooks/tree/master/sagemaker) |
|
- [Amazon SageMaker documentation for Hugging Face](https://docs.aws.amazon.com/sagemaker/latest/dg/hugging-face.html) |
|
- [Python SDK SageMaker documentation for Hugging Face](https://sagemaker.readthedocs.io/en/stable/frameworks/huggingface/index.html) |
|
- [Deep Learning Container](https://github.com/aws/deep-learning-containers/blob/master/available_images.md#huggingface-training-containers) |
|
|
|
## Result |
|
| key | value | |
|
| --- | ----- | |
|
| key | value | |
|
| key | value | |
|
|
|
## Usage |
|
```python |
|
from transformers import pipeline |
|
summarizer = pipeline("summarization", model="philschmid/bart-base-samsum") |
|
|
|
conversation = '''Jeff: Can I train a π€ Transformers model on Amazon SageMaker? |
|
Philipp: Sure you can use the new Hugging Face Deep Learning Container. |
|
Jeff: ok. |
|
Jeff: and how can I get started? |
|
Jeff: where can I find documentation? |
|
Philipp: ok, ok you can find everything here. https://huggingface.co/blog/the-partnership-amazon-sagemaker-and-hugging-face |
|
''' |
|
nlp(conversation) |
|
``` |
|
|