Spaces:
Runtime error
Runtime error
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
|