Update app.py

app.py

@@ -1,6 +1,13 @@
 import gradio as gr
 from util import update
 
+def get_model(cp):
+    checkpoint = cp
+    tokenizer, model = load_model(checkpoint)
+    return tokenizer, model
+
+tokenizer, model = get_model(cp_aug)
+
 with gr.Blocks() as demo:
     gr.Markdown("Start typing below and then click **Run** to see the output.")
     with gr.Row():