fix: set top k as 1

app.py

@@ -29,7 +29,7 @@ def augment(img, transforms=transforms) -> torch.Tensor:
     return transforms(img).unsqueeze(0)
 
 
-def search_index(input_image, k: int):
+def search_index(input_image, k = 1):
     with torch.no_grad():
         embedding = model(augment(input_image))
         index = read_index("./bin/dino.index")