Create text_summarize.py

text_summarize.py

@@ -0,0 +1,14 @@
+import torch 
+import gradio as gr 
+
+from transformers import pipeline
+
+text_summary = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6", torch_dtype=torch.bfloat16)
+
+def summarize (text):
+    output = text_summary(text)
+    return output[0]['summary_text']
+
+gr.close_all()
+app = gr.Interface(fn=summarize,inputs=gr.Textbox(label="Plug in your text here to summarize", lines=10, placeholder="Enter text to summarize..."), outputs=gr.Textbox(label="Summary", lines=5, placeholder="Summary will appear here..."), title="Text-Summarizer", description="This application summarizes the input text.", theme="default")
+app.launch()