jusancp99 commited on
Commit
1c64895
1 Parent(s): 97f393c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -22,7 +22,10 @@ lsh_builder.lsh = loaded_lsh
22
 
23
  # Candidate images.
24
  dataset = load_dataset("gjuggler/bird-data")
25
- candidate_dataset = dataset["train"].shuffle(seed=seed)
 
 
 
26
 
27
  def query(image, top_k):
28
  results = lsh_builder.query(image)
 
22
 
23
  # Candidate images.
24
  dataset = load_dataset("gjuggler/bird-data")
25
+ num_samples = 10000
26
+ seed = 42
27
+ candidate_dataset= dataset1["train"].shuffle(seed=seed).select(range(num_samples))
28
+ #candidate_dataset = dataset["train"].shuffle(seed=seed)
29
 
30
  def query(image, top_k):
31
  results = lsh_builder.query(image)