yay!

app.py

@@ -25,11 +25,8 @@ def query(image, top_k):
     model_output = model(**inputs)
     embedding = model_output.pooler_output.detach()
     results = index.query(embedding)
-    images=[]
-    for i in results[0].tolist():
-    	print(i)
-    	print(type(i))
-    	images.append(ds.select(i)["image"])
+    inx = results[0][0].tolist()
+    images = ds.select(inx)["image"]
     return images