add rag search

ragatouille_search.py

@@ -1,9 +1,10 @@
 from pathlib import Path
+from typing import Any, Dict, List
+
 import gradio as gr
 from huggingface_hub import snapshot_download
 from ragatouille import RAGPretrainedModel
 from toolz import unique
-from typing import List, Dict, Any
 
 # Top-level variables
 INDEX_PATH = Path(".ragatouille/colbert/indexes/my_index_with_ids_and_metadata/")
@@ -81,9 +82,9 @@ def create_ragatouille_interface():
             Feel free to open a discussion to give feedback or request features 🤗
             """
         )
-
-        query = gr.Textbox(label="Search query", placeholder="medieval handwriting")
         with gr.Column():
+            query = gr.Textbox(label="Search query", placeholder="medieval handwriting")
+        with gr.Row():
             k = gr.Slider(1, 100, value=DEFAULT_K, step=1, label="Number of Results")
             make_unique = gr.Checkbox(False, label="Show each dataset only once?")
         search_button = gr.Button("Search")