yes2code commited on
Commit
7d83082
·
1 Parent(s): b4d0042

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -6,6 +6,10 @@ import gradio as gr
6
  #iface = gr.Interface(fn=greet, inputs="text", outputs="text")
7
  #iface.launch()
8
 
 
 
 
 
9
  def summarize(input):
10
  output = get_completion(input)
11
  return output[0]['summary_text']
 
6
  #iface = gr.Interface(fn=greet, inputs="text", outputs="text")
7
  #iface.launch()
8
 
9
+ from transformers import pipeline
10
+
11
+ get_completion = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6")
12
+
13
  def summarize(input):
14
  output = get_completion(input)
15
  return output[0]['summary_text']