Update app.py

changing to 1 image

app.py

@@ -18,7 +18,7 @@ dataset = load_dataset("sasha/butterflies_10k_names_multiple")
 ds = dataset["train"]
 
 
-def query(image, top_k=4):
+def query(image, top_k=1):
     inputs = feature_extractor(image, return_tensors="pt")
     model_output = model(**inputs)
     embedding = model_output.pooler_output.detach()