Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 from util import update
+from util import summarize
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, DataCollatorForSeq2Seq, Seq2SeqTrainer, TrainingArguments
 
 cp_aug = 'minnehwg/finetune-newwiki-summarization-ver-augmented'
@@ -7,12 +8,14 @@ cp_aug = 'minnehwg/finetune-newwiki-summarization-ver-augmented'
 tokenizer = AutoTokenizer.from_pretrained("VietAI/vit5-base")
 model  = AutoModelForSeq2SeqLM.from_pretrained(cp_aug)
 
+
+
 with gr.Blocks() as demo:
     gr.Markdown("Start typing below and then click **Run** to see the output.")
     with gr.Row():
         inp = gr.Textbox(placeholder="What is your name?")
         out = gr.Textbox()
     btn = gr.Button("Run")
-    btn.click(fn=update, inputs=inp, outputs=out)
+    btn.click(fn=summarize, inputs=inp, outputs=out)
 
 demo.launch()