nkasmanoff commited on
Commit
3d598a4
1 Parent(s): cb516f1

Update dataset_recommender.py

Browse files
Files changed (1) hide show
  1. 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 = OpenAIEmbeddings()):
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