Update app.py

app.py

@@ -1,13 +1,26 @@
+from util import pipeline
+from util import load_model
 import gradio as gr
-def update(name):
-    return f"Welcome to Gradio, {name}!"
-
-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)
-
-demo.launch()
+
+cp_aug = 'minnehwg/finetune-newwiki-summarization-ver-augmented2'
+
+def get_model():
+    checkpoint = cp_aug 
+    tokenizer, model = load_model(checkpoint)
+    return tokenizer, model
+
+tokenizer, model = get_model()
+
+def generate_summary(input_text):
+    return pipeline(input_text)
+
+demo = gr.Interface(
+    fn=generate_summary,
+    inputs=gr.inputs.Textbox(lines=2, placeholder="Enter your URL..."),
+    outputs=gr.outputs.Textbox(label="Generated Text"),
+    title="Chào mừng đến với hệ thống tóm tắt của Minne >.<",
+    description="Enter the URL to summarize and click 'Submit' to generate the summary."
+)
+
+if __name__ == "__main__":
+    demo.launch()