Spaces:
Runtime error
Runtime error
nkasmanoff
commited on
Commit
•
3d598a4
1
Parent(s):
cb516f1
Update dataset_recommender.py
Browse files- dataset_recommender.py +1 -1
dataset_recommender.py
CHANGED
@@ -10,7 +10,7 @@ from langchain.embeddings import OpenAIEmbeddings
|
|
10 |
class DatasetRecommender:
|
11 |
def __init__(self, dataset = 'nkasmanoff/huggingface-datasets' ,
|
12 |
llm_backbone = ChatOpenAI(),
|
13 |
-
embeddings_backbone =
|
14 |
self.dataset = dataset
|
15 |
self.llm_backbone = llm_backbone
|
16 |
self.embeddings_backbone = embeddings_backbone
|
|
|
10 |
class DatasetRecommender:
|
11 |
def __init__(self, dataset = 'nkasmanoff/huggingface-datasets' ,
|
12 |
llm_backbone = ChatOpenAI(),
|
13 |
+
embeddings_backbone = HuggingFaceEmbeddings()):
|
14 |
self.dataset = dataset
|
15 |
self.llm_backbone = llm_backbone
|
16 |
self.embeddings_backbone = embeddings_backbone
|