Update app.py

app.py

@@ -29,7 +29,7 @@ def summ_t5(text):
     return summary_text
     
 def summ_bert(text):
-    input_ids = tokenizer_bert(text, return_tensors="pt")
+    input_ids = tokenizer_bert.encode(text, return_tensors="pt")
     summary_ids= model_bert.generate(input_ids,
                 max_length=100,
                 num_beams=10,
@@ -43,7 +43,7 @@ def summ_bert(text):
     return summary_text
 
 def para_t5(text):
-    encoding = t5_para_tokenizer.encode(text, padding='longest', return_tensors='pt')
+    encoding = t5_para_tokenizer(text, padding='longest', return_tensors='pt')
     outputs = t5_para_model.generate(
                 input_ids=encoding["input_ids"],
                 attention_mask=encoding["attention_mask"],
@@ -61,10 +61,10 @@ def para_t5(text):
     ]
 
 def summarize(text):
-    #t5_ = summ_t5(text)
+    t5_ = summ_t5(text)
     bert_ = summ_bert(text)
-    #para_ = para_t5(t5_)
-    return bert_
+    para_ = para_t5(t5_)
+    return t5_, bert_, para_
 
 if __name__ == "__main__":
     with gr.Blocks() as demo:
@@ -73,8 +73,8 @@ if __name__ == "__main__":
                 input_text = gr.Textbox(label="Input Text")
                 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")
-                #para_output = gr.Textbox(label="T5 Paraphrase Output")
-        analyze_button.click(summarize, inputs=input_text, outputs=[bert_output])
+                t5_output = gr.Textbox(label="T5 Base Output")
+                bert_output = gr.Textbox(label="Bert2Bert Base Output")
+                para_output = gr.Textbox(label="T5 Paraphrase Output")
+        analyze_button.click(summarize, inputs=input_text, outputs=[t5_ouput, bert_output, para_output])
     demo.launch()