Update app.py

app.py

@@ -31,7 +31,7 @@ processor = ColQwen2Processor.from_pretrained("manu/colqwen2-v1.0-alpha")
 
 @spaces.GPU
 def search(query: str, ds, images, k):
-
+    k = min(k, len(ds))
     device = "cuda:0" if torch.cuda.is_available() else "cpu"
     if device != model.device:
         model.to(device)