Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -35,7 +35,8 @@ from langchain_google_genai import GoogleGenerativeAIEmbeddings
|
|
35 |
|
36 |
|
37 |
# LLM: HuggingFace
|
38 |
-
from
|
|
|
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 =
|
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 |
|