Update README.md

README.md

@@ -19,24 +19,33 @@ Finetuned EncoderDecoder model using BERT-base and GPT2-small for Indonesian tex
 ## How to Load Finetuned Model
 
 ```python
-from transformers import BertTokenizer, EncoderDecoderModel
+from transformers import BertTokenizer, EncoderDecoderModel, EncoderDecoderConfig
 model_ckpt = 'ardavey/bert2gpt-indosum'
 tokenizer = BertTokenizer.from_pretrained(model_ckpt)
 tokenizer.bos_token = tokenizer.cls_token
 tokenizer.eos_token = tokenizer.sep_token
-model = EncoderDecoderModel.from_pretrained(model_ckpt)
+
+config = EncoderDecoderConfig.from_pretrained(model_ckpt)
+config.early_stopping = True
+
+model = EncoderDecoderModel.from_pretrained(model_ckpt, config=config)
 ```
 
 ## Code Example
 
 ```python
-model_path = 'ardavey/bert2gpt-indosum'
-tokenizer = BertTokenizer.from_pretrained(model_path)
+from transformers import BertTokenizer, EncoderDecoderModel, EncoderDecoderConfig
+model_ckpt = 'ardavey/bert2gpt-indosum'
+tokenizer = BertTokenizer.from_pretrained(model_ckpt)
 tokenizer.bos_token = tokenizer.cls_token
 tokenizer.eos_token = tokenizer.sep_token
-model = EncoderDecoderModel.from_pretrained(model_path)
 
-# Source of the article below: [](https://soc.telkomuniversity.ac.id/optiguard-aplikasi-dan-alat-untuk-deteksi-dini-penyakit-kebutaan-ciptaan-mahasiswa-fakultas-informatika-tel-u/)
+config = EncoderDecoderConfig.from_pretrained(model_ckpt)
+config.early_stopping = True
+
+model = EncoderDecoderModel.from_pretrained(model_ckpt, config=config)
+
+# Source of the article below: https://soc.telkomuniversity.ac.id/optiguard-aplikasi-dan-alat-untuk-deteksi-dini-penyakit-kebutaan-ciptaan-mahasiswa-fakultas-informatika-tel-u
 custom_text = """
 Bandung, 5 Juli 2024 --- Kebutaan akan membawa dampak negatif bagi penderitanya. Pentingnya menjaga Kesehatan mata untuk mencegah kebutaan membuat sekelompok mahasiswa dari Fakultas Informatika Telkom University menciptakan inovasi untuk mengidentifikasi penyakit mata.
 Inovasi ini diberi nama Optiguard yang dikembangkan melalui Program Kreativitas Mahasiswa Bidang Karsa Cipta (PKM-KC) dan berhasil mendapatkan pendanaan dari Kemendikbudristek. OptiGuard diciptakan sebagai upaya preventif mencegah kebutaan dengan mendeteksi dini penyakit mata, sehingga penyakit mata dapat ditangani sebelum semakin parah dan memperkecil peluang terjadinya kebutaan. Inovasi OptiGuard ini mengedepakan efisiensi, kecepatan, dan ketepatan dalam mendiagnosis penyakit mata.