Update app.py

app.py

@@ -6,6 +6,11 @@ from gradio.mix import Parallel
 tokenizer_t5 = T5Tokenizer.from_pretrained("panggi/t5-base-indonesian-summarization-cased")
 model_t5 = T5ForConditionalGeneration.from_pretrained("panggi/t5-base-indonesian-summarization-cased")
 
+tokenizer_bert = BertTokenizer.from_pretrained("cahya/bert2bert-indonesian-summarization")
+tokenizer_bert.bos_token = tokenizer.cls_token
+tokenizer_bert.eos_token = tokenizer.sep_token
+model_bert = EncoderDecoderModel.from_pretrained("cahya/bert2bert-indonesian-summarization")
+
 def summ_t5(text):
     input_ids = tokenizer_t5.encode(text, return_tensors='pt')
     summary_ids = model_t5.generate(input_ids,
@@ -18,14 +23,37 @@ def summ_t5(text):
                 use_cache=True)
     summary_text = tokenizer_t5.decode(summary_ids[0], skip_special_tokens=True)
     return summary_text
+    
+def summ_bert(text):
+    input_ids = tokenizer_bert.encode(ARTICLE_TO_SUMMARIZE, return_tensors='pt')
+    summary_ids = model_bert.generate(input_ids,
+                min_length=20,
+                max_length=100,
+                num_beams=10,
+                repetition_penalty=2.5,
+                length_penalty=1.0,
+                early_stopping=True,
+                no_repeat_ngram_size=2,
+                use_cache=True,
+                do_sample = True,
+                temperature = 0.8,
+                top_k = 50,
+                top_p = 0.95)
+    summary_text = tokenizer_bert.decode(summary_ids[0], skip_special_tokens=True)
+    return summary_text
 
-summ_demo = gr.Interface(
+t5_demo = gr.Interface(
     fn=summ_t5,
     inputs="text",
     outputs=gr.Textbox(lines=10, label="T5 Base Output")
     )
+bert_demo = gr.Interface(
+    fn=summ_bert,
+    inputs="text",
+    outputs=gr.Textbox(lines=10, label="Bert2Bert Base Output")
+    )
 
 if __name__ == "__main__":
-    Parallel(summ_demo,
+    Parallel(t5_demo, bert_demo,
     inputs=gr.Textbox(lines=10, label="Input Text", placeholder="Enter article here..."),
     title="Summary of Summarizer - Indonesia").launch()