nkasmanoff commited on
Commit
bfc72f4
1 Parent(s): 14d6d06

Update dataset_recommender.py

Browse files
Files changed (1) hide show
  1. dataset_recommender.py +1 -1
dataset_recommender.py CHANGED
@@ -8,7 +8,7 @@ from langchain.embeddings import OpenAIEmbeddings
8
 
9
 
10
  class DatasetRecommender:
11
- def __init__(self, dataset ,
12
  llm_backbone = ChatOpenAI(),
13
  embeddings_backbone = OpenAIEmbeddings()):
14
  self.dataset = dataset
 
8
 
9
 
10
  class DatasetRecommender:
11
+ def __init__(self, dataset = 'nkasmanoff/huggingface-datasets' ,
12
  llm_backbone = ChatOpenAI(),
13
  embeddings_backbone = OpenAIEmbeddings()):
14
  self.dataset = dataset