yilmazmusa commited on
Commit
53da45d
1 Parent(s): 0dd9926

fixed app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -7,12 +7,14 @@ import torch
7
  from transformers import pipeline, AutoProcessor, AutoModel
8
  import numpy as np
9
  import gradio as gr
 
10
  from scipy.io.wavfile import write as write_wav
11
  warnings.filterwarnings("ignore")
12
 
13
 
14
  # Here is the code
15
  def extract_abstract(uploaded_file):
 
16
  with pdfplumber.open(pdf_bytes) as pdf:
17
  abstract = ""
18
  # Iterate through each page
@@ -85,7 +87,7 @@ def summarize_and_convert_to_audio(pdf_file):
85
  # Create a Gradio interface
86
  iface = gr.Interface(
87
  fn=summarize_and_convert_to_audio,
88
- inputs=gr.UploadButton(label="Upload PDF", type="filepath", file_types=["pdf"]), # Set to accept only PDF files
89
  outputs=gr.Audio(label="Audio"),
90
  title="PDF Abstract Summarizer",
91
  description="Upload a PDF with an abstract to generate a summarized audio."
 
7
  from transformers import pipeline, AutoProcessor, AutoModel
8
  import numpy as np
9
  import gradio as gr
10
+ from io import BytesIO
11
  from scipy.io.wavfile import write as write_wav
12
  warnings.filterwarnings("ignore")
13
 
14
 
15
  # Here is the code
16
  def extract_abstract(uploaded_file):
17
+ pdf_bytes = BytesIO(uploaded_file)
18
  with pdfplumber.open(pdf_bytes) as pdf:
19
  abstract = ""
20
  # Iterate through each page
 
87
  # Create a Gradio interface
88
  iface = gr.Interface(
89
  fn=summarize_and_convert_to_audio,
90
+ inputs=gr.UploadButton(label="Upload PDF", type="binary", file_types=["pdf"]), # Set to accept only PDF files
91
  outputs=gr.Audio(label="Audio"),
92
  title="PDF Abstract Summarizer",
93
  description="Upload a PDF with an abstract to generate a summarized audio."