tosin2013 commited on
Commit
0c0ed04
·
verified ·
1 Parent(s): 21adb3e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -24,12 +24,13 @@ def answer_question(model_name, file, question, status):
24
  model = load_model(model_name)
25
 
26
  if file is not None:
27
- if file.type == "application/pdf":
 
28
  pdf_reader = PyPDF2.PdfFileReader(file)
29
  context = ""
30
  for page_num in range(pdf_reader.numPages):
31
  context += pdf_reader.getPage(page_num).extract_text()
32
- elif file.name.endswith(".md"):
33
  context = file.read().decode('utf-8')
34
  context = markdown.markdown(context)
35
  else:
 
24
  model = load_model(model_name)
25
 
26
  if file is not None:
27
+ file_name = file.name
28
+ if file_name.endswith(".pdf"):
29
  pdf_reader = PyPDF2.PdfFileReader(file)
30
  context = ""
31
  for page_num in range(pdf_reader.numPages):
32
  context += pdf_reader.getPage(page_num).extract_text()
33
+ elif file_name.endswith(".md"):
34
  context = file.read().decode('utf-8')
35
  context = markdown.markdown(context)
36
  else: