Update app.py

app.py

@@ -6,13 +6,13 @@ 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_bert.cls_token
-#tokenizer_bert.eos_token = tokenizer_bert.sep_token
-#model_bert = EncoderDecoderModel.from_pretrained("cahya/bert2bert-indonesian-summarization")
+tokenizer_bert = BertTokenizer.from_pretrained("cahya/bert2bert-indonesian-summarization")
+tokenizer_bert.bos_token = tokenizer_bert.cls_token
+tokenizer_bert.eos_token = tokenizer_bert.sep_token
+model_bert = EncoderDecoderModel.from_pretrained("cahya/bert2bert-indonesian-summarization")
 
-t5_para_tokenizer = AutoTokenizer.from_pretrained("Wikidepia/IndoT5-base-paraphrase")  
-t5_para_model = AutoModelForSeq2SeqLM.from_pretrained("Wikidepia/IndoT5-base-paraphrase")
+#t5_para_tokenizer = AutoTokenizer.from_pretrained("Wikidepia/IndoT5-base-paraphrase")  
+#t5_para_model = AutoModelForSeq2SeqLM.from_pretrained("Wikidepia/IndoT5-base-paraphrase")
 
 
 def summ_t5(text):
@@ -31,18 +31,13 @@ def summ_t5(text):
 def summ_bert(text):
     input_ids = tokenizer_bert(text, 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)
+                use_cache=True)
 
     summary_text = tokenizer_bert.decode(summary_ids[0], skip_special_tokens=True)
     return summary_text
@@ -67,9 +62,9 @@ def para_t5(text):
 
 def summarize(text):
     t5_ = summ_t5(text)
-    #bert_ = summ_bert(text)
+    bert_ = summ_bert(text)
     #para_ = para_t5(t5_)
-    return t5_
+    return t5_, bert_
 
 if __name__ == "__main__":
     with gr.Blocks() as demo:
@@ -79,7 +74,7 @@ if __name__ == "__main__":
                 analyze_button = gr.Button(label="Analyze")
             with gr.Column():
                 t5_output = gr.Textbox(label="T5 Base Output")
-                #bert_output = gr.Textbox(label="Bert Base Output")
+                bert_output = gr.Textbox(label="Bert Base Output")
                 #para_output = gr.Textbox(label="T5 Paraphrase Output")
-        analyze_button.click(summarize, input_text, [t5_output])
+        analyze_button.click(summarize, inputs=input_text, outputs=[t5_output, bert_output])
     demo.launch()