Waflon commited on
Commit
c24dee1
1 Parent(s): 9cdcee6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -8
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