update readme
Browse files
README.md
CHANGED
|
@@ -5,6 +5,8 @@ language:
|
|
| 5 |
- id
|
| 6 |
base_model:
|
| 7 |
- cahya/bert2bert-indonesian-summarization
|
|
|
|
|
|
|
| 8 |
---
|
| 9 |
|
| 10 |
# Fine-Tuned BERT2BERT Summarization Model
|
|
@@ -33,7 +35,7 @@ ARTICLE = ""
|
|
| 33 |
# generate summary
|
| 34 |
input_ids = tokenizer.encode(ARTICLE, return_tensors='pt')
|
| 35 |
summary_ids = model.generate(input_ids,
|
| 36 |
-
max_length=
|
| 37 |
num_beams=2,
|
| 38 |
repetition_penalty=2.5,
|
| 39 |
length_penalty=1.0,
|
|
|
|
| 5 |
- id
|
| 6 |
base_model:
|
| 7 |
- cahya/bert2bert-indonesian-summarization
|
| 8 |
+
library: transformers
|
| 9 |
+
pipeline_tag: Summarization
|
| 10 |
---
|
| 11 |
|
| 12 |
# Fine-Tuned BERT2BERT Summarization Model
|
|
|
|
| 35 |
# generate summary
|
| 36 |
input_ids = tokenizer.encode(ARTICLE, return_tensors='pt')
|
| 37 |
summary_ids = model.generate(input_ids,
|
| 38 |
+
max_length=125,
|
| 39 |
num_beams=2,
|
| 40 |
repetition_penalty=2.5,
|
| 41 |
length_penalty=1.0,
|