Update app.py

app.py

@@ -27,7 +27,7 @@ def query(image, top_k=4):
     logits = results[1][0].tolist()
     images = ds.select(inx)["image"]
     captions = ds.select(inx)["label"]
-    images_with_captions = [(i, c) for i, c in zip(images, captions)]
+    images_with_captions = [(i, ds.features["label"].int2str(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()}
     try: