MusIre commited on
Commit
c6a6aba
·
1 Parent(s): 3cf581d

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +34 -0
app.py ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import subprocess
2
+
3
+ subprocess.run(["pip", "install", "PyPDF2", "transformers", "bark", "gradio"])
4
+
5
+ import PyPDF2
6
+ from transformers import pipeline
7
+ from bark import SAMPLE_RATE, generate_audio, preload_models
8
+ import gradio as gr
9
+
10
+ def summarize_and_convert_to_audio(pdf_path, abstract_page):
11
+ with open(pdf_path, 'rb') as file:
12
+ pdf_reader = PyPDF2.PdfReader(file)
13
+
14
+ # Get the abstract page text
15
+ abstract_page_text = pdf_reader.pages[abstract_page - 1].extract_text()
16
+
17
+ summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
18
+ summary = summarizer(abstract_page_text, max_length=20, min_length=20)
19
+
20
+ preload_models()
21
+
22
+ text = summary[0]['summary_text']
23
+ audio_array = generate_audio(text)
24
+
25
+ return Audio(audio_array, rate=SAMPLE_RATE)
26
+
27
+ iface = gr.Interface(
28
+ fn=summarize_and_convert_to_audio,
29
+ inputs=["file", "number"],
30
+ outputs="audio",
31
+ live=True
32
+ )
33
+
34
+ iface.launch()