|
|
--- |
|
|
license: mit |
|
|
datasets: |
|
|
- Samsung/samsum |
|
|
language: |
|
|
- en |
|
|
metrics: |
|
|
- rouge |
|
|
library_name: transformers |
|
|
pipeline_tag: summarization |
|
|
--- |
|
|
|
|
|
|
|
|
Model obtained by Fine Tuning 'google/pegasus-cnn_dailymail' using SAMSUM Dataset! |
|
|
## Usage |
|
|
# Example 1 |
|
|
```python |
|
|
from transformers import pipeline |
|
|
summarizer = pipeline("summarization", model="neuronstarml/samsuntextsum") |
|
|
text = ''' |
|
|
Hannah: Hey, do you have Betty's number? |
|
|
Amanda: Lemme check |
|
|
Hannah: <file_gif> |
|
|
Amanda: Sorry, can't find it. |
|
|
Amanda: Ask Larry |
|
|
Amanda: He called her last time we were at the park together |
|
|
Hannah: I don't know him well |
|
|
Hannah: <file_gif> |
|
|
Amanda: Don't be shy, he's very nice |
|
|
Hannah: If you say so.. |
|
|
Hannah: I'd rather you texted him |
|
|
Amanda: Just text him 🙂 |
|
|
Hannah: Urgh.. Alright |
|
|
Hannah: Bye |
|
|
Amanda: Bye bye |
|
|
''' |
|
|
summarizer(text) |
|
|
``` |
