Create app.py

app.py

@@ -1,51 +1,12 @@
-
-from util import load_model
-from util import pipeline
-
 import gradio as gr
-cp_aug = 'minnehwg/finetune-newwiki-summarization-ver-augmented2'
-
-
-def get_model(cp):
-    checkpoint = cp
-    tokenizer, model = load_model(checkpoint)
-    return tokenizer, model
-
-tokenizer, model = get_model(cp_aug)
-
-
-def generate_summary(url):
-    results = pipeline(url, model, tokenizer)
-    summary = "\n".join(results)
-    return summary
-
-def generate_summary_and_video(url):
-    summary = generate_summary(url)
-    summary_html = summary.replace("\n", "<br>")
-    try:
-        video_id = url.split("v=")[1].split("&")[0]
-        iframe = f'<iframe width="300" height="200" src="https://www.youtube.com/embed/{video_id}" frameborder="0" allow="accelerometer; autoplay; encrypted-media; gyroscope; picture-in-picture" allowfullscreen></iframe>'
-        return f"{iframe}<br><br>Những ý chính trong video:<br><br>{summary_html}"
-    except IndexError:
-        return f"**Summary:**\n{summary}\n\nInvalid YouTube URL for video display."
-
-css = """
-.output-html {
-    font-size: 40px;
-}
-"""
-
-demo = gr.Interface(
-    fn=generate_summary_and_video,
-    inputs=gr.Textbox(lines=2, placeholder="Enter URL..."),
-    outputs=gr.HTML(label="Results"),
-    title="Summarizer",
-    description="Enter the URL to display the YouTube video and summarize the content.",
-    css=css
-)
-
-
-demo.launch(share=True)
-
+from util import update
 
+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()