vsrinivas commited on
Commit
e7e12d3
·
1 Parent(s): 008c811

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -1,6 +1,9 @@
1
  import gradio as gr
2
  gr.close_all()
3
 
 
 
 
4
  def summarize(input):
5
  output = get_completion(input)
6
  return output[0]['summary_text']
 
1
  import gradio as gr
2
  gr.close_all()
3
 
4
+ from transformers import pipeline
5
+ get_completion = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6")
6
+
7
  def summarize(input):
8
  output = get_completion(input)
9
  return output[0]['summary_text']