Update app.py

app.py

@@ -6,10 +6,10 @@ 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")
@@ -66,9 +66,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_, bert_, para_
+    return t5_, para_
 
 if __name__ == "__main__":
     with gr.Blocks() as demo:
@@ -78,7 +78,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, bert_output, para_output])
+        analyze_button.click(summarize, input_text, [t5_output, para_output])
     demo.launch()