Update app.py

app.py

@@ -1,18 +1,26 @@
 import gradio as gr
+from transformers import pipeline
 
-api = gr.Interface.load("huggingface/EleutherAI/gpt-j-6B")
-
-
-def complete_with_gpt(text):
-    # Use the last 50 characters of the text as context
-    return text[:-50] + api(text[-50:])
+pipe = pipeline("summarization", model="Gabriel/bart-base-cnn-xsum-swe")
 
+def main(in_text):
+  print(in_text)
+  answer = pipe(in_text, num_beams=5 ,min_length=5, max_length=20)
+  print(answer)
+  return answer[0]["summary_text"]
 
 with gr.Blocks() as demo:
-    textbox = gr.Textbox(placeholder="Type here and press enter...", lines=4)
-    btn = gr.Button("Generate")
-
-    btn.click(complete_with_gpt, textbox, textbox)
+  gr.Markdown("""# Summarization Engine!""")
+  with gr.Row():
+    with gr.Column():
+      text1 = gr.Textbox(
+            label="Input Text",
+            lines=1,
+        )
+      output = gr.Textbox(label="Output Text")
+      b1 = gr.Button("Summarize!")
+      b1.click(main, inputs=[text1], outputs=output)
+  gr.Markdown("""#### Created by Gabriel""")
 
 
 if __name__ == "__main__":