Update app.py

app.py

@@ -21,10 +21,15 @@ except Exception as e:
     st.stop()
 
 # 3. 问答函数
-def answer_question(gemma, temp, max, question):
+def answer_question(repo_id, temperature, max_length, question):
     # 4. 初始化 Gemma 模型
     try:
-        llm = HuggingFaceEndpoint(repo_id=gemma, temperature=temp, max_length=max)
+        llm = HuggingFaceEndpoint(
+            repo_id=repo_id, 
+            temperature=temperature, 
+            max_length=max_length,
+            huggingfacehub_api_token=HUGGINGFACEHUB_API_TOKEN
+        )
     except Exception as e:
         st.error(f"Gemma 模型加载失败：{e}")
         st.stop()
@@ -51,7 +56,7 @@ def answer_question(gemma, temp, max, question):
 # 6. Streamlit 界面
 st.title("Gemma 知识库问答系统")
 
-gemma = st.selectbox("repo-id", ("google/gemma-2-9b-it", "google/gemma-2-2b-it", "google/recurrentgemma-2b-it"), 2)
+gemma = st.selectbox("repo-id", ("google/gemma-7b-it", "google/gemma-2b-it", "google/recurrentgemma-2b-it"), 2)
 temperature = st.number_input("temperature", value=1.0)
 max_length = st.number_input("max_length", value=1024)
 question = st.text_area("请输入问题", "Gemma 有哪些特点？")