maviced commited on
Commit
2dac5b5
1 Parent(s): 2aa38ae

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -26,7 +26,7 @@ lsh_builder = BuildLSHTable(model)
26
  lsh_builder.lsh = loaded_lsh
27
 
28
  # Candidate images.
29
- dataset = load_dataset("beans")
30
  candidate_dataset = dataset["train"].shuffle(seed=seed)
31
 
32
 
@@ -57,7 +57,7 @@ def query(image, top_k):
57
  return list(zip(images, labels))
58
 
59
 
60
- title = "Fetch Similar Beans 🪴"
61
  description = "This Space demos an image similarity system. You can refer to [this notebook](TODO) to know the details of the system. You can pick any image from the available samples below. On the right hand side, you'll find the similar images returned by the system. The example images have been named with their corresponding integer class labels for easier identification. The fetched images will also have their integer labels tagged so that you can validate the correctness of the results."
62
 
63
  # You can set the type of gr.Image to be PIL, numpy or str (filepath)
 
26
  lsh_builder.lsh = loaded_lsh
27
 
28
  # Candidate images.
29
+ dataset = load_dataset("Matthijs/snacks")
30
  candidate_dataset = dataset["train"].shuffle(seed=seed)
31
 
32
 
 
57
  return list(zip(images, labels))
58
 
59
 
60
+ title = "Fetch Similar Snacks 🪴"
61
  description = "This Space demos an image similarity system. You can refer to [this notebook](TODO) to know the details of the system. You can pick any image from the available samples below. On the right hand side, you'll find the similar images returned by the system. The example images have been named with their corresponding integer class labels for easier identification. The fetched images will also have their integer labels tagged so that you can validate the correctness of the results."
62
 
63
  # You can set the type of gr.Image to be PIL, numpy or str (filepath)