Update app.py

app.py

@@ -5,12 +5,12 @@ from util import pipeline
 import gradio as gr
 cp_aug = 'minnehwg/finetune-newwiki-summarization-ver-augmented2'
 
-def get_model():
-    checkpoint = cp_aug 
+def get_model(cp):
+    checkpoint = cp
     tokenizer, model = load_model(checkpoint)
     return tokenizer, model
 
-tokenizer, model = get_model()
+tokenizer, model = get_model(cp_aug)
 
 # demo = gr.Interface(
 #     fn=generate_summary,
@@ -20,16 +20,21 @@ tokenizer, model = get_model()
 #     description="Enter the URL to summarize and click 'Submit' to generate the summary."
 # )
 
-def generate(text):
-    return pipeline(text, model, tokenizer)
+def generate_summary(url):
+    summary = pipeline(url, model, tokenizer)
+    return summary
 
-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="Enter your URL")
-        out = gr.Textbox()
-    btn = gr.Button("Run")
-    btn.click(fn=generate, inputs=inp, outputs=out)
+# Tạo interface Gradio với hàm generate_summary
+demo = gr.Interface(
+    fn=generate_summary,
+    inputs=gr.Textbox(lines=2, placeholder="Enter your URL..."),
+    outputs=gr.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."
+)
 
+# Chạy interface
 demo.launch()
 
+
+