Spaces:
Sleeping
Sleeping
Commit
·
53da45d
1
Parent(s):
0dd9926
fixed app.py
Browse files
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="
|
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."
|