Update app.py

app.py

@@ -1,16 +1,10 @@
-import gradio as gr 
 from PyPDF2 import PdfReader
 from transformers import pipeline
-# from bark import SAMPLE_RATE, generate_audio, preload_models
-from IPython.display import Audio
+import gradio as gr
 
-# Preload Bark models
-preload_models()
-
-def summarize_generate_audio(files):
+def summarize_pdf_content(files):
     summarizer = pipeline("summarization", model="stevhliu/my_awesome_billsum_model")
     summaries = []
-#    audios = []
 
     for file in files:
         with open(file.name, "rb") as pdf_file:
@@ -20,18 +14,13 @@ def summarize_generate_audio(files):
                 text += page.extract_text() + "\n"
             summary = summarizer(text, max_length=50, min_length=10, do_sample=False)[0]['summary_text']
             summaries.append(summary)
-#            audio_array = generate_audio(summary)
-#            audios.append(audio_array)
-    return summaries, #audios
+    return summaries
 
 iface = gr.Interface(
-    fn=summarize_generate_audio,
+    fn=summarize_pdf_content,
     inputs=gr.UploadButton("Click to Upload a PDF", file_types=["pdf"], file_count="multiple"),
-    outputs=[
-        gr.Textbox(label="Summarized Text"),
-#        gr.Audio(label="Generated Audio")
-    ],
-    title="PDF Summarizer & Audio Generator"
+    outputs=gr.Textbox(label="Summarized Text"),
+    title="PDF Summarizer"
 )
 
-iface.launch() 
+iface.launch()