Merge pull request #141 from arc53/demo

demo on docsgpt docs
2 years ago · adb912be06
parent d651077e4e b6772d8358
commit adb912be06
3 changed files with 5 additions and 1 deletions
--- a/application/app.py
+++ b/application/app.py
@ -136,7 +136,11 @@ def api_answer():
        # some formatting for the frontend
        result['answer'] = result['result']
        result['answer'] = result['answer'].replace("\\n", "<br>")
-        result['answer'] = result['answer'].replace("SOURCES:", "")
+        try:
+            result['answer'] = result['answer'].split("SOURCES:")[0]
+        except:
+            pass
+
        # mock result
        # result = {
        #     "answer": "The answer is 42",
--- a/application/index.faiss
+++ b/application/index.faiss
--- a/application/index.pkl
+++ b/application/index.pkl