varl42 commited on
Commit
37c818f
1 Parent(s): d8d3df1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -22,7 +22,8 @@ def summarize_text(text):
22
  break
23
  abstract = ". ".join(sentences[start:end+1])
24
  summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
25
- summary = summarizer(abstract, max_length=50, min_length=30)
 
26
  return summary[0]['summary_text']
27
 
28
  def text_to_audio(text):
@@ -40,12 +41,14 @@ def audio_pdf(pdf_file):
40
 
41
  inputs = gr.File()
42
  audio_summary = gr.Audio()
 
43
 
44
  iface = gr.Interface(
45
  fn=audio_pdf,
46
  inputs=inputs,
47
- outputs=audio_summary,
48
- title="PDF Summarizer"
 
49
  )
50
 
51
  iface.launch()
 
22
  break
23
  abstract = ". ".join(sentences[start:end+1])
24
  summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
25
+ summary = summarizer(abstract, max_length=50, min_length=30,
26
+ do_sample=False)
27
  return summary[0]['summary_text']
28
 
29
  def text_to_audio(text):
 
41
 
42
  inputs = gr.File()
43
  audio_summary = gr.Audio()
44
+ summary_text = gr.Text()
45
 
46
  iface = gr.Interface(
47
  fn=audio_pdf,
48
  inputs=inputs,
49
+ outputs=[summary_text,audio_summary]
50
+ title="PDF Summarizer",
51
+ examples=["Attention_is_all_you_need.pdf", "ImageNet_Classification.pdf"]
52
  )
53
 
54
  iface.launch()