ryanrwatkins commited on
Commit
501ff3f
1 Parent(s): ce631b1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -180,8 +180,8 @@ langchain_document_loader()
180
 
181
  text_splitter = RecursiveCharacterTextSplitter(
182
  separators = ["\n\n", "\n", " ", ""],
183
- chunk_size = 2000,
184
- chunk_overlap= 20
185
  )
186
 
187
  # Text splitting
@@ -228,7 +228,8 @@ def select_embeddings_model(LLM_service="HuggingFace"):
228
  if LLM_service == "HuggingFace":
229
  embeddings = HuggingFaceInferenceAPIEmbeddings(
230
  api_key=HF_key,
231
- model_name="thenlper/gte-large"
 
232
  )
233
  print("embedding model selected")
234
  return embeddings
 
180
 
181
  text_splitter = RecursiveCharacterTextSplitter(
182
  separators = ["\n\n", "\n", " ", ""],
183
+ chunk_size = 1500,
184
+ chunk_overlap= 50
185
  )
186
 
187
  # Text splitting
 
228
  if LLM_service == "HuggingFace":
229
  embeddings = HuggingFaceInferenceAPIEmbeddings(
230
  api_key=HF_key,
231
+ #model_name="thenlper/gte-large"
232
+ model_name="sentence-transformers/all-MiniLM-l6-v2"
233
  )
234
  print("embedding model selected")
235
  return embeddings