Update app.py

app.py

@@ -230,7 +230,7 @@ def search_synthetize(query: str, k: int = 5) -> List[int]:
     """
     top_k_indices = search(query, k)
     expanded = set(top_k_indices)
-    for i in base:
+    for i in top_k_indices:
         expanded.add(i - 1)
         expanded.add(i + 1)
     expanded = {i for i in expanded if 0 <= i < len(images)}