Commit
·
de6402a
1
Parent(s):
f282230
Update README.md
Browse files
README.md
CHANGED
@@ -34,6 +34,20 @@ This model is a specialized adaptation of the <b>facebook/bart-large-xsum</b>, f
|
|
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 |
## Training Parameters
|
38 |
```python
|
39 |
evaluation_strategy = "epoch",
|
@@ -51,18 +65,4 @@ num_train_epochs=4,
|
|
51 |
predict_with_generate=True,
|
52 |
fp16=True,
|
53 |
report_to="none"
|
54 |
-
```
|
55 |
-
|
56 |
-
## Usage
|
57 |
-
|
58 |
-
```python
|
59 |
-
from transformers import pipeline
|
60 |
-
|
61 |
-
model = pipeline("summarization", model="luisotorres/bart-finetuned-samsum")
|
62 |
-
|
63 |
-
conversation = '''Sarah: Do you think it's a good idea to invest in Bitcoin?
|
64 |
-
Emily: I'm skeptical. The market is very volatile, and you could lose money.
|
65 |
-
Sarah: True. But there's also a high upside, right?
|
66 |
-
'''
|
67 |
-
model(conversation)
|
68 |
```
|
|
|
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 |
+
from transformers import pipeline
|
41 |
+
|
42 |
+
model = pipeline("summarization", model="luisotorres/bart-finetuned-samsum")
|
43 |
+
|
44 |
+
conversation = '''Sarah: Do you think it's a good idea to invest in Bitcoin?
|
45 |
+
Emily: I'm skeptical. The market is very volatile, and you could lose money.
|
46 |
+
Sarah: True. But there's also a high upside, right?
|
47 |
+
'''
|
48 |
+
model(conversation)
|
49 |
+
```
|
50 |
+
|
51 |
## Training Parameters
|
52 |
```python
|
53 |
evaluation_strategy = "epoch",
|
|
|
65 |
predict_with_generate=True,
|
66 |
fp16=True,
|
67 |
report_to="none"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
68 |
```
|