luisotorres
commited on
Commit
•
852d4ac
1
Parent(s):
7fb0238
Update README.md
Browse files
README.md
CHANGED
@@ -1,3 +1,50 @@
|
|
1 |
---
|
2 |
pipeline_tag: summarization
|
3 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
---
|
2 |
pipeline_tag: summarization
|
3 |
+
datasets:
|
4 |
+
- samsum
|
5 |
+
language:
|
6 |
+
- en
|
7 |
+
metrics:
|
8 |
+
- rouge
|
9 |
+
library_name: transformers
|
10 |
+
widget:
|
11 |
+
- text: |
|
12 |
+
John: Hey! I've been thinking about getting a PlayStation 5. Do you think it is worth it?
|
13 |
+
Dan: Idk man. R u sure ur going to have enough free time to play it?
|
14 |
+
John: Yeah, that's why I'm not sure if I should buy one or not. I've been working so much lately idk if I'm gonna be able to play it as much as I'd like.
|
15 |
+
- text: |
|
16 |
+
Sarah: Do you think it's a good idea to invest in Bitcoin?
|
17 |
+
Emily: I'm skeptical. The market is very volatile, and you could lose money.
|
18 |
+
Sarah: True. But there's also a high upside, right?
|
19 |
+
- text: |
|
20 |
+
Madison: Hello Lawrence are you through with the article?
|
21 |
+
Lawrence: Not yet sir.
|
22 |
+
Lawrence: But i will be in a few.
|
23 |
+
Madison: Okay. But make it quick.
|
24 |
+
Madison: The piece is needed by today
|
25 |
+
Lawrence: Sure thing
|
26 |
+
Lawrence: I will get back to you once i am through."
|
27 |
+
|
28 |
+
---
|
29 |
+
|
30 |
+
# Description
|
31 |
+
|
32 |
+
This model is a specialized adaptation of the <b>facebook/bart-large-xsum</b>, fine-tuned for enhanced performance on dialogue summarization using the <b>SamSum</b> dataset.
|
33 |
+
|
34 |
+
## Development
|
35 |
+
- Kaggle Notebook: [Text Summarization with Large Language Models](https://www.kaggle.com/code/lusfernandotorres/text-summarization-with-large-language-models)
|
36 |
+
|
37 |
+
## Usage
|
38 |
+
|
39 |
+
```python
|
40 |
+
|
41 |
+
from transformers import pipeline
|
42 |
+
|
43 |
+
model = pipeline("summarization", model="luisotorres/bart-finetuned-samsum")
|
44 |
+
|
45 |
+
conversation = '''Sarah: Do you think it's a good idea to invest in Bitcoin?
|
46 |
+
Emily: I'm skeptical. The market is very volatile, and you could lose money.
|
47 |
+
Sarah: True. But there's also a high upside, right?
|
48 |
+
'''
|
49 |
+
model(conversation)
|
50 |
+
```
|