Update app.py
Browse files
app.py
CHANGED
|
@@ -2,6 +2,7 @@
|
|
| 2 |
|
| 3 |
import streamlit as st
|
| 4 |
import os
|
|
|
|
| 5 |
import json
|
| 6 |
from dotenv import load_dotenv
|
| 7 |
from haystack.nodes.prompt import PromptNode, PromptTemplate
|
|
@@ -82,7 +83,6 @@ def get_refs(res):
|
|
| 82 |
Parse response for engineered reference ids (refer to prompt template)
|
| 83 |
Extract documents using reference ids
|
| 84 |
'''
|
| 85 |
-
import re
|
| 86 |
text = res["results"][0]
|
| 87 |
# This pattern should be returned by gpt3.5
|
| 88 |
# pattern = r'ref\. (\d+)\]\.'
|
|
@@ -103,7 +103,7 @@ def run_query(input_text):
|
|
| 103 |
st.success(output)
|
| 104 |
references = get_refs(res)
|
| 105 |
st.write('References')
|
| 106 |
-
st.success(references)
|
| 107 |
|
| 108 |
|
| 109 |
# Setup retriever, pulling from local faiss datastore
|
|
|
|
| 2 |
|
| 3 |
import streamlit as st
|
| 4 |
import os
|
| 5 |
+
import re
|
| 6 |
import json
|
| 7 |
from dotenv import load_dotenv
|
| 8 |
from haystack.nodes.prompt import PromptNode, PromptTemplate
|
|
|
|
| 83 |
Parse response for engineered reference ids (refer to prompt template)
|
| 84 |
Extract documents using reference ids
|
| 85 |
'''
|
|
|
|
| 86 |
text = res["results"][0]
|
| 87 |
# This pattern should be returned by gpt3.5
|
| 88 |
# pattern = r'ref\. (\d+)\]\.'
|
|
|
|
| 103 |
st.success(output)
|
| 104 |
references = get_refs(res)
|
| 105 |
st.write('References')
|
| 106 |
+
st.success(str(references))
|
| 107 |
|
| 108 |
|
| 109 |
# Setup retriever, pulling from local faiss datastore
|