Merge branch 'main' of https://huggingface.co/panggi/t5-small-indonesian-summarization-cased into main

README.md

@@ -32,13 +32,13 @@ ARTICLE_TO_SUMMARIZE = "Secara umum, dispepsia adalah kumpulan gejala pada salur
 # generate summary
 input_ids = tokenizer.encode(ARTICLE_TO_SUMMARIZE, return_tensors='pt')
 summary_ids = model.generate(input_ids,
-                                max_length=100,
-                                do_sample=True,
-                                no_repeat_ngram_size=2,
-                                early_stopping=True,
-                                temperature = 0.8,
-                                top_k = 50,
-                                top_p = 0.95)
+            max_length=100, 
+            num_beams=2,
+            repetition_penalty=2.5, 
+            length_penalty=1.0, 
+            early_stopping=True,
+            no_repeat_ngram_size=2,
+            use_cache=True)
 summary_text = tokenizer.decode(summary_ids[0], skip_special_tokens=True)
 print(summary_text)
 ```