changing col name

app.py

@@ -6,7 +6,7 @@ import wikipedia
 
 
 # Only runs once when the script is first run.
-with open("insectarium_768.pickle", "rb") as handle:
+with open("./insectarium_768.pickle", "rb") as handle:
     index = pickle.load(handle)
 
 # Load model for computing embeddings.
@@ -26,7 +26,7 @@ def query(image, top_k=4):
     inx = results[0][0].tolist()
     logits = results[1][0].tolist()
     images = ds.select(inx)["image"]
-    captions = ds.select(inx)["name"]
+    captions = ds.select(inx)["label"]
     images_with_captions = [(i, c) for i, c in zip(images, captions)]
     labels_with_probs = dict(zip(captions, logits))
     labels_with_probs = {k: 1 - v for k, v in labels_with_probs.items()}