Charles Chan commited on
Commit
4b8fc6b
·
1 Parent(s): 742b788
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -63,6 +63,7 @@ def get_answer(prompt):
63
  answer = st.session_state.llm.invoke(prompt)
64
  # 去掉 prompt 的内容
65
  answer = answer.replace(prompt, "").strip()
 
66
  st.success("答案已经生成!")
67
  print("答案已经生成!")
68
  return answer
@@ -105,7 +106,7 @@ def answer_question(repo_id, temperature, max_length, question):
105
  return {"prompt": prompt, "answer": answer, "pure_answer": pure_answer}
106
  except Exception as e:
107
  st.error(f"问答过程出错:{e}")
108
- return {"prompt": "", "answer": "An error occurred during the answering process."}
109
 
110
  # Streamlit 界面
111
  st.title("進擊的巨人 知识库问答系统")
 
63
  answer = st.session_state.llm.invoke(prompt)
64
  # 去掉 prompt 的内容
65
  answer = answer.replace(prompt, "").strip()
66
+ print(answer)
67
  st.success("答案已经生成!")
68
  print("答案已经生成!")
69
  return answer
 
106
  return {"prompt": prompt, "answer": answer, "pure_answer": pure_answer}
107
  except Exception as e:
108
  st.error(f"问答过程出错:{e}")
109
+ return {"prompt": "", "answer": "An error occurred during the answering process.", "pure_answer": ""}
110
 
111
  # Streamlit 界面
112
  st.title("進擊的巨人 知识库问答系统")