firstAPP / app.py
masakk's picture
Update app.py
830896f
# https://huggingface.co/spaces/masakk/firstAPP
from gtts import gTTS
from pdfminer.high_level import extract_text
from transformers import pipeline
import gradio as gr
import os
summarizer=(pipeline(task='summarization'))
def find_abstract(input_text):
count=0
for item in input_text.split("\n\n"):
count=count+1
if item.lower()=="abstract":
break
return count
def pdf_to_text(file_obj):
text = extract_text(file_obj.name)
summaryPDF=summarizer(text.split("\n\n")[find_abstract(text)], max_length=20, min_length=5, do_sample=False)
myobj = gTTS(text=summaryPDF[0]["summary_text"], lang='en', slow=False)
myobj.save("test.wav")
return 'test.wav'
iface = gr.Interface(fn = pdf_to_text,
inputs = 'file',
outputs = 'audio',
title = 'PDF to Audio Application',
description = 'A simple application to convert PDF files in audio speech. Upload your own file, it must contain abstract.',
article =
'''<div>
<p style="text-align: center"> All you need to do is to upload the pdf file and hit submit, then wait for compiling. After that click on Play/Pause for listing to the audio. The audio is saved in a wav format.</p>
</div>''',
)
iface.launch()