Update app.py

app.py

@@ -4,12 +4,8 @@ from util import load_model
 
 cp_aug = 'minnehwg/finetune-newwiki-summarization-ver-augmented'
 
-def get_model(cp):
-    checkpoint = cp
-    tokenizer, model = load_model(checkpoint)
-    return tokenizer, model
-
-tokenizer, model = get_model(cp_aug)
+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.")