rockerritesh commited on
Commit
b6a8855
·
verified ·
1 Parent(s): 75bc1c1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -5
app.py CHANGED
@@ -148,12 +148,11 @@ def main():
148
  file_extension = os.path.splitext(uploaded_file.name)[1].lower()
149
 
150
  if file_extension == ".pdf":
151
- pdf_reader = PyPDF2.PdfReader(io.BytesIO(uploaded_file.read()))
152
- text = ""
153
- for page in pdf_reader.pages:
154
- text += page.extract_text()
155
  else: # .txt file
156
- text = uploaded_file.getvalue().decode("utf-8")
 
157
 
158
  converted_text = convert(text)
159
 
 
148
  file_extension = os.path.splitext(uploaded_file.name)[1].lower()
149
 
150
  if file_extension == ".pdf":
151
+ text = extract_text_from_pdf(io.BytesIO(uploaded_file.read()))
152
+
 
 
153
  else: # .txt file
154
+ with open(inputfile, "r") as fp:
155
+ text = fp.read()
156
 
157
  converted_text = convert(text)
158