Update app.py

app.py

@@ -70,9 +70,11 @@ def qa_result(context, question, file, pipe = pipe):
             elif extension == allowed_types[1]:
                 context = extract_text_from_txt(file.name)
             else:
-                context = extract_text_from_doc(file.name)
+                context = extract_text_from_pdf(file.name)
                 result = pipe(question=question, context=context)
                 answered = result['answer']
+                text = remove_references(answered)
+
             # text = remove_references(answered)
    
     if file is None and len(context) == 0 and len(question) == 0:
@@ -88,8 +90,8 @@ def qa_result(context, question, file, pipe = pipe):
         else:
             result = pipe(question=question, context=context)
             answered = result['answer']
+            text = remove_references(answered)
     
-    text = remove_references(answered)
     text = text.replace('(', '', 1)
     text = text.replace(',', '', len(text)-1)