cbir-image-similarity

Runtime error

Samuel Schmidt commited on Mar 7, 2023

Commit

0cbab2a

1 Parent(s): a329e3c

Update dataset size

Files changed (1) hide show

src/app.py CHANGED Viewed

@@ -6,7 +6,7 @@ import numpy as np
 from datasets import *
 dataset = load_dataset("huggan/CelebA-faces")
-candidate_subset = dataset["train"].select(range(40)) # This is a small CBIR app! :D
 def emb_dataset(dataset):
     # This function might need to be split up, to reduce start-up time of app

 from datasets import *
 dataset = load_dataset("huggan/CelebA-faces")
+candidate_subset = dataset["train"].select(range(1000)) # This is a small CBIR app! :D
 def emb_dataset(dataset):
     # This function might need to be split up, to reduce start-up time of app