Update app.py

app.py

@@ -1,8 +1,25 @@
 import gradio as gr
 
+from src.TextSummarizer.pipeline.prediction import PredictionPipeline
 
-def greet(name):
-    return "Hello " + name + "!!"
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
-demo.launch()
+def predict(document):
+    """
+    Method will take the document and summarize it.
+    """
+
+    # predict the summary using my own pre-trained model.
+    summary = PredictionPipeline().predict(document)
+    return summary
+
+
+# Create the frontend.
+input_interfaces: list = []
+
+with gr.Blocks(theme=gr.theme.Soft()) as app:
+    with gr.Row():
+        title = "Text Summarizer..."
+
+
+
+app.launch()