luisotorres commited on
Commit
852d4ac
1 Parent(s): 7fb0238

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +48 -1
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
+ ```