RanjithkumarPanjabikesan commited on
Commit
73a6969
·
verified ·
1 Parent(s): a034f1f

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +18 -0
app.py ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #Summarize any text using the `shleifer/distilbart-cnn-12-6` model
2
+ #Create a text summarization Pipeline
3
+ from transformers import pipeline
4
+ tsp = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6")
5
+ #Build the Text Summarization App and Launch
6
+ import gradio as gr
7
+ #Define summarization function
8
+ def summarize(input):
9
+ output = tsp(input)
10
+ return output[0]['summary_text']
11
+
12
+ textsummarizationapp = gr.Interface(fn=summarize,
13
+ inputs=[gr.Textbox(label="Text to summarize", lines=6)],
14
+ outputs=[gr.Textbox(label="Result", lines=3)],
15
+ title="Text summarization with distilbart-cnn",
16
+ description="Summarize any text using the `sshleifer/distilbart-cnn-12-6` model"
17
+ )
18
+ textsummarizationapp.launch()