Add dataset

app.py

@@ -11,7 +11,7 @@ title = r"""
 
 description = r"""
 
-In this demo, we use [DiffusionDB](https://huggingface.co/datasets/poloclub/diffusiondb) instead of [LAION](https://laion.ai/blog/laion-400-open-dataset/) because LAION is currently not available. 
+In this demo, we use subset of [danbooru22](https://huggingface.co/datasets/animelover/danbooru2022) or [DiffusionDB](https://huggingface.co/datasets/poloclub/diffusiondb) instead of [LAION](https://laion.ai/blog/laion-400-open-dataset/) instead of [LAION](https://laion.ai/blog/laion-400-open-dataset/) because LAION is currently not available. 
 <br>
 This demo currently supports text search only.
 <br>
@@ -21,6 +21,7 @@ The code is based on [clip-retrieval](https://github.com/rom1504/clip-retrieval)
 
 """
 
+#In this demo, we use  because LAION is currently not available. 
 # From local file
 # INDEX_DIR = "dataset/diffusiondb/text_index_folder"
 # IND = faiss.read_index(f"{INDEX_DIR}/text.index")
@@ -105,7 +106,7 @@ with gr.Blocks() as demo:
     gr.Markdown(description)
 
     with gr.Row():
-        dataset = gr.Dropdown(label="dataset", choices=["DiffusionDB"], value="DiffusionDB")
+        dataset = gr.Dropdown(label="dataset", choices=["booru22", "DiffusionDB"], value="DiffusionDB")
         top_k = gr.Slider(label="top k", minimum=1, maximum=20, value=8)
         show_score = gr.Checkbox(label="Show score", value=True)
     query_text = gr.Textbox(label="query text")