Fix outdated code

app.py

@@ -19,6 +19,8 @@ DESCRIPTION = """
 Image similarity-based retrieval tool using:
 - [SmilingWolf/wd-v1-4-convnext-tagger-v2](https://huggingface.co/SmilingWolf/wd-v1-4-convnext-tagger-v2) as feature extractor
 - [Faiss](https://github.com/facebookresearch/faiss) and [autofaiss](https://github.com/criteo/autofaiss) for indexing
+
+Also, check out [SmilingWolf/danbooru2022_embeddings_playground](https://huggingface.co/spaces/SmilingWolf/danbooru2022_embeddings_playground) for a similar space with experimental support for text input combined with image input.
 """
 
 HF_TOKEN = os.environ["HF_TOKEN"]
@@ -183,12 +185,15 @@ def main():
                         visible=False,
                     )
                     n_neighbours = gr.Slider(
-                        minimum=1, maximum=20, value=5, step=1, label="# of images"
+                        minimum=1,
+                        maximum=20,
+                        value=5,
+                        step=1,
+                        label="# of images",
                     )
                 find_btn = gr.Button("Find similar images")
-        similar_images = gr.Gallery(label="Similar images")
+        similar_images = gr.Gallery(label="Similar images", columns=[5])
 
-        similar_images.style(grid=5)
         find_btn.click(
             fn=searcher.predict,
             inputs=[