trying like this

app.py

@@ -3,9 +3,7 @@ import gradio as gr
 from datasets import load_dataset
 from transformers import AutoModel, AutoFeatureExtractor
 import wikipedia
-from platform import python_version
 
-print(python_version())
 
 # Only runs once when the script is first run.
 with open("butts_1024_new.pickle", "rb") as handle:
@@ -29,23 +27,9 @@ def query(image, top_k=1):
     results = index.query(embedding, k=top_k)
     inx = results[0][0].tolist()
     logits = results[1][0].tolist()
-    images = ds.select(inx)["image"]
-    captions = ds.select(inx)["name"]
-    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()}
-    try:
-        description = wikipedia.summary(captions[0], sentences=1)
-        description = "### " + description
-        url = wikipedia.page(captions[0]).url
-        url = " You can learn more about your butterfly [here](" + str(url) + ")!"
-        description = description + url
-    except:
-        description = "### Butterflies are insects in the order Lepidoptera, which also includes moths. Adult butterflies have large, often brightly coloured wings."
-        url = "https://en.wikipedia.org/wiki/Butterfly"
-        url = " You can learn more about butterflies [here](" + str(url) + ")!"
-        description = description + url
-    return images_with_captions, labels_with_probs, description
+    butterfly = ds.select(inx)["image"][0]
+    butterfly.show()
+    return butterfly
 
 
 with gr.Blocks() as demo:
@@ -61,7 +45,6 @@ with gr.Blocks() as demo:
 
         with gr.Column(scale=2):
             outputs = gr.Gallery(rows=1)
-            labels = gr.Label()
 
     gr.Markdown("### Image Examples")
     gr.Examples(
@@ -71,6 +54,6 @@ with gr.Blocks() as demo:
         fn=query,
         cache_examples=True,
     )
-    btn.click(query, inputs, [outputs, labels, description])
+    btn.click(query, inputs, butterfly)
 
 demo.launch()