Charles Chan
commited on
Commit
·
5ce5698
1
Parent(s):
51c0f15
coding
Browse files
app.py
CHANGED
@@ -5,6 +5,9 @@ from langchain_community.embeddings import SentenceTransformerEmbeddings
|
|
5 |
from langchain_community.vectorstores import FAISS
|
6 |
from datasets import load_dataset
|
7 |
|
|
|
|
|
|
|
8 |
# 使用 假知识 数据集
|
9 |
if "data_list" not in st.session_state:
|
10 |
st.session_state.data_list = []
|
@@ -17,7 +20,7 @@ if not st.session_state.data_list:
|
|
17 |
data_list = []
|
18 |
answer_list = []
|
19 |
for example in dataset["train"]:
|
20 |
-
answer_list.append(example["Answer"])
|
21 |
data_list.append({"Question": example["Question"], "Answer": example["Answer"]})
|
22 |
st.session_state.answer_list = answer_list
|
23 |
st.session_state.data_list = data_list
|
@@ -106,9 +109,6 @@ def answer_question(repo_id, temperature, max_length, question):
|
|
106 |
st.error(f"问答过程出错:{e}")
|
107 |
return {"prompt": "", "answer": "An error occurred during the answering process.", "pure_answer": ""}
|
108 |
|
109 |
-
# Streamlit 界面
|
110 |
-
st.title("假知识库问答系统")
|
111 |
-
|
112 |
col1, col2 = st.columns(2)
|
113 |
with col1:
|
114 |
gemma = st.selectbox("repo-id", ("google/gemma-2-9b-it", "google/gemma-2-2b-it", "google/recurrentgemma-2b-it"), 2)
|
|
|
5 |
from langchain_community.vectorstores import FAISS
|
6 |
from datasets import load_dataset
|
7 |
|
8 |
+
# Streamlit 界面
|
9 |
+
st.title("假知识库问答系统")
|
10 |
+
|
11 |
# 使用 假知识 数据集
|
12 |
if "data_list" not in st.session_state:
|
13 |
st.session_state.data_list = []
|
|
|
20 |
data_list = []
|
21 |
answer_list = []
|
22 |
for example in dataset["train"]:
|
23 |
+
answer_list.append(example["Question"] + example["Answer"])
|
24 |
data_list.append({"Question": example["Question"], "Answer": example["Answer"]})
|
25 |
st.session_state.answer_list = answer_list
|
26 |
st.session_state.data_list = data_list
|
|
|
109 |
st.error(f"问答过程出错:{e}")
|
110 |
return {"prompt": "", "answer": "An error occurred during the answering process.", "pure_answer": ""}
|
111 |
|
|
|
|
|
|
|
112 |
col1, col2 = st.columns(2)
|
113 |
with col1:
|
114 |
gemma = st.selectbox("repo-id", ("google/gemma-2-9b-it", "google/gemma-2-2b-it", "google/recurrentgemma-2b-it"), 2)
|