Charles Chan commited on
Commit
5ce5698
·
1 Parent(s): 51c0f15
Files changed (1) hide show
  1. app.py +4 -4
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)