Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
|
|
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
|
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:
|