Spaces:
Running
Running
Update app.py
Browse files
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 |
-
|
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 =
|
239 |
api_key=HF_key,
|
240 |
-
model_name="thenlper/gte-large",
|
241 |
-
|
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 |
|