zeerd commited on
Commit
b6628bb
·
verified ·
1 Parent(s): a06a315

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -52,10 +52,10 @@ def answer_question(gemma, temperature, max_length, question):
52
  # 5. Streamlit 界面
53
  st.title("Gemma 知识库问答系统")
54
 
55
- gemma = st.text_area("模型", "google/gemma-7b-it", height=50)
56
- temperature = st.text_area("temperature", "1.0", height=50)
57
- max_length = st.text_area("max_length", "1024", height=50)
58
- question = st.text_area("请输入问题", "Gemma 有哪些特点?", height=100)
59
 
60
  if st.button("提交"):
61
  if not question:
 
52
  # 5. Streamlit 界面
53
  st.title("Gemma 知识库问答系统")
54
 
55
+ gemma = st.selectbox("模型", ("google/gemma-2-9b-it", "google/gemma-2-2b-it", "google/recurrentgemma-2b-it"), 2)
56
+ temperature = st.text_area("temperature", "1.0")
57
+ max_length = st.text_area("max_length", "1024")
58
+ question = st.text_area("请输入问题", "Gemma 有哪些特点?")
59
 
60
  if st.button("提交"):
61
  if not question: