Update app.py
Browse files
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()
|