Update app.py

app.py

@@ -5,9 +5,15 @@ from langchain_community.embeddings.sentence_transformer import SentenceTransfor
 from langchain_community.vectorstores import Chroma
 import streamlit as st
 
-embedding_function = SentenceTransformerEmbeddings(model_name="all-mpnet-base-v2")
-chdb = Chroma(persist_directory="./chroma_db_info", embedding_function=embedding_function)
-
+lang = st.radio("Choose Language:",
+               ["English", "中文"])
+if lang == "English":
+    embedding_function = SentenceTransformerEmbeddings(model_name="all-mpnet-base-v2")
+    chdb = Chroma(persist_directory="./chroma_db_info", embedding_function=embedding_function)
+else:
+    embedding_function = SentenceTransformerEmbeddings(model_name="shibing624/text2vec-base-chinese")
+    chdb = Chroma(persist_directory="./chroma_db_ch", embedding_function=embedding_function)
+   
 query = st.text_input("Enter a query")
 disnum = 3
 if query: