Update app.py

app.py

@@ -17,7 +17,7 @@ ds.add_faiss_index(column='embeddings')
 def query(image, number_to_retrieve=1):
     input_image = model.encode(image)
     scores, retrieved_examples = ds.get_nearest_examples('embeddings', input_image, k=number_to_retrieve)
-    return retrieved_examples['image']
+    return retrieved_examples['image'][0]
 
 
 with gr.Blocks() as demo: