Update app.py

app.py

@@ -4,9 +4,12 @@ from langchain_community.embeddings import SentenceTransformerEmbeddings
 from langchain.vectorstores import FAISS
 import numpy as np
 
+# gemma = 'google/gemma-7b-it';
+gemma = 'google/recurrentgemma-2b-it';
+
 # 1. 初始化 Gemma 模型
 try:
-    llm = HuggingFaceHub(repo_id="google/gemma-7b-it", model_kwargs={"temperature": 0.5, "max_length": 512})
+    llm = HuggingFaceHub(repo_id=gemma, model_kwargs={"temperature": 0.5, "max_length": 512})
 except Exception as e:
     st.error(f"Gemma 模型加载失败：{e}")
     st.stop()