JPLTedCas commited on
Commit
082997b
·
verified ·
1 Parent(s): eac4b8a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -131,6 +131,7 @@ def get_vectorstore(text_chunks):
131
  MODEL_NAME = "WhereIsAI/UAE-Large-V1"
132
  MODEL_NAME = "sentence-transformers/all-MiniLM-L6-v2"
133
  MODEL_NAME = "avsolatorio/GIST-Embedding-v0"
 
134
 
135
  hf_embeddings = HuggingFaceEmbeddings(model_name=MODEL_NAME)
136
  vectorstore = Chroma.from_documents(text_chunks, hf_embeddings, persist_directory="db")
 
131
  MODEL_NAME = "WhereIsAI/UAE-Large-V1"
132
  MODEL_NAME = "sentence-transformers/all-MiniLM-L6-v2"
133
  MODEL_NAME = "avsolatorio/GIST-Embedding-v0"
134
+ MODEL_NAME = "Salesforce/SFR-Embedding-Mistral"
135
 
136
  hf_embeddings = HuggingFaceEmbeddings(model_name=MODEL_NAME)
137
  vectorstore = Chroma.from_documents(text_chunks, hf_embeddings, persist_directory="db")