ryanrwatkins commited on
Commit
5096b7c
·
verified ·
1 Parent(s): 241a40a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -35,8 +35,8 @@ from langchain_google_genai import GoogleGenerativeAIEmbeddings
35
 
36
 
37
  # LLM: HuggingFace
38
- from langchain_huggingface.embeddings import HuggingFaceEmbeddings
39
- #from langchain_community.embeddings import HuggingFaceInferenceAPIEmbeddings
40
  from langchain_community.llms import HuggingFaceHub
41
 
42
  # langchain prompts, memory, chains...
@@ -235,10 +235,10 @@ def select_embeddings_model(LLM_service="HuggingFace"):
235
  """
236
 
237
  if LLM_service == "HuggingFace":
238
- embeddings = HuggingFaceEmbeddings(
239
  api_key=HF_key,
240
- model_name="thenlper/gte-large",
241
- #model_name="sentence-transformers/all-MiniLM-L6-v2",
242
  #model_name="sentence-transformers/all-mpnet-base-v2",
243
  )
244
 
 
35
 
36
 
37
  # LLM: HuggingFace
38
+ #from langchain_huggingface.embeddings import HuggingFaceEmbeddings
39
+ from langchain_community.embeddings import HuggingFaceInferenceAPIEmbeddings
40
  from langchain_community.llms import HuggingFaceHub
41
 
42
  # langchain prompts, memory, chains...
 
235
  """
236
 
237
  if LLM_service == "HuggingFace":
238
+ embeddings = HuggingFaceInferenceAPIEmbeddings(
239
  api_key=HF_key,
240
+ #model_name="thenlper/gte-large",
241
+ model_name="sentence-transformers/all-MiniLM-L6-v2",
242
  #model_name="sentence-transformers/all-mpnet-base-v2",
243
  )
244