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)
|