Update app.py
Browse files
app.py
CHANGED
|
@@ -6,7 +6,8 @@ from langchain_community.vectorstores import Chroma
|
|
| 6 |
import streamlit as st
|
| 7 |
|
| 8 |
lang = st.radio("Choose Language:",
|
| 9 |
-
["English", "中文"]
|
|
|
|
| 10 |
if lang == "English":
|
| 11 |
embedding_function = SentenceTransformerEmbeddings(model_name="all-mpnet-base-v2")
|
| 12 |
chdb = Chroma(persist_directory="./chroma_db_info", embedding_function=embedding_function)
|
|
@@ -14,7 +15,7 @@ else:
|
|
| 14 |
embedding_function = SentenceTransformerEmbeddings(model_name="shibing624/text2vec-base-chinese")
|
| 15 |
chdb = Chroma(persist_directory="./chroma_db_ch", embedding_function=embedding_function)
|
| 16 |
|
| 17 |
-
query = st.text_input("Enter a query")
|
| 18 |
disnum = 3
|
| 19 |
if query:
|
| 20 |
docs = chdb.similarity_search_with_score(query)
|
|
|
|
| 6 |
import streamlit as st
|
| 7 |
|
| 8 |
lang = st.radio("Choose Language:",
|
| 9 |
+
["English", "中文"],
|
| 10 |
+
captions= ['Support in English', 'Chinese QnA'])
|
| 11 |
if lang == "English":
|
| 12 |
embedding_function = SentenceTransformerEmbeddings(model_name="all-mpnet-base-v2")
|
| 13 |
chdb = Chroma(persist_directory="./chroma_db_info", embedding_function=embedding_function)
|
|
|
|
| 15 |
embedding_function = SentenceTransformerEmbeddings(model_name="shibing624/text2vec-base-chinese")
|
| 16 |
chdb = Chroma(persist_directory="./chroma_db_ch", embedding_function=embedding_function)
|
| 17 |
|
| 18 |
+
query = st.text_input("Enter a query about IC design:")
|
| 19 |
disnum = 3
|
| 20 |
if query:
|
| 21 |
docs = chdb.similarity_search_with_score(query)
|