Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -79,7 +79,8 @@ def get_vectorstore(text_chunks):
|
|
| 79 |
A FAISS vector store containing the embeddings of the text chunks.
|
| 80 |
|
| 81 |
"""
|
| 82 |
-
model = "BAAI/bge-base-en-v1.5"
|
|
|
|
| 83 |
encode_kwargs = {
|
| 84 |
"normalize_embeddings": True
|
| 85 |
} # set True to compute cosine similarity
|
|
@@ -150,15 +151,8 @@ def main():
|
|
| 150 |
|
| 151 |
st.write(css, unsafe_allow_html=True)
|
| 152 |
|
| 153 |
-
# set huggingface hub token in st.text_input widget
|
| 154 |
-
# then hide the input
|
| 155 |
-
huggingface_token = st.text_input("Enter your HuggingFace Hub token", type="password")
|
| 156 |
-
#openai_api_key = st.text_input("Enter your OpenAI API key", type="password")
|
| 157 |
-
|
| 158 |
# set this key as an environment variable
|
| 159 |
os.environ["HUGGINGFACEHUB_API_TOKEN"] = huggingface_token
|
| 160 |
-
#os.environ["OPENAI_API_KEY"] = openai_api_key
|
| 161 |
-
|
| 162 |
|
| 163 |
if "conversation" not in st.session_state:
|
| 164 |
st.session_state.conversation = None
|
|
|
|
| 79 |
A FAISS vector store containing the embeddings of the text chunks.
|
| 80 |
|
| 81 |
"""
|
| 82 |
+
#model = "BAAI/bge-base-en-v1.5"
|
| 83 |
+
model = "sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2"
|
| 84 |
encode_kwargs = {
|
| 85 |
"normalize_embeddings": True
|
| 86 |
} # set True to compute cosine similarity
|
|
|
|
| 151 |
|
| 152 |
st.write(css, unsafe_allow_html=True)
|
| 153 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 154 |
# set this key as an environment variable
|
| 155 |
os.environ["HUGGINGFACEHUB_API_TOKEN"] = huggingface_token
|
|
|
|
|
|
|
| 156 |
|
| 157 |
if "conversation" not in st.session_state:
|
| 158 |
st.session_state.conversation = None
|