Jerome Blin commited on
Commit
0638a02
1 Parent(s): d74a522

feat: use full dataset in app

Browse files
Files changed (1) hide show
  1. fis/app/app.py +1 -1
fis/app/app.py CHANGED
@@ -18,7 +18,7 @@ PIPELINE_NAME = "dummy_swin_pipe"
18
 
19
  pipeline = factory.get(PIPELINE_NAME)
20
 
21
- DATASET_PATH = os.path.join(ORGANISATION, "dummy_swin_pipe_5k")
22
  dataset = load_dataset(path=DATASET_PATH, split="train")
23
  dataset.add_faiss_index(column="embedding")
24
 
 
18
 
19
  pipeline = factory.get(PIPELINE_NAME)
20
 
21
+ DATASET_PATH = os.path.join(ORGANISATION, PIPELINE_NAME)
22
  dataset = load_dataset(path=DATASET_PATH, split="train")
23
  dataset.add_faiss_index(column="embedding")
24