ryanrwatkins commited on
Commit
59b77b1
·
verified ·
1 Parent(s): 4908b55

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -4
app.py CHANGED
@@ -35,7 +35,8 @@ from langchain_google_genai import GoogleGenerativeAIEmbeddings
35
 
36
 
37
  # LLM: HuggingFace
38
- from langchain_community.embeddings import HuggingFaceInferenceAPIEmbeddings
 
39
  from langchain_community.llms import HuggingFaceHub
40
 
41
  # langchain prompts, memory, chains...
@@ -234,11 +235,14 @@ def select_embeddings_model(LLM_service="HuggingFace"):
234
  """
235
 
236
  if LLM_service == "HuggingFace":
237
- embeddings = HuggingFaceInferenceAPIEmbeddings(
238
  api_key=HF_key,
239
- model_name="thenlper/gte-large"
240
- #model_name="sentence-transformers/all-MiniLM-L6-v2"
 
241
  )
 
 
242
  print(embeddings)
243
  return embeddings
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 = 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
+
245
+
246
  print(embeddings)
247
  return embeddings
248