Allow enter to work on the Textbox

app.py

@@ -28,7 +28,7 @@ model = SentenceTransformer(
 def search(query, top_k: int = 10, rerank_multiplier: int = 4):
     # 1. Embed the query as float32
     start_time = time.time()
-    query_embedding = model.encode(query)
+    query_embedding = model.encode([query])
     embed_time = time.time() - start_time
 
     # 2. Quantize the query to ubinary
@@ -76,10 +76,11 @@ with gr.Blocks(title="Quantized Retrieval") as demo:
 
     with gr.Row():
         with gr.Column(scale=4):
-            output = gr.Dataframe(column_widths=["10%", "20%", "80%"], headers=["Score", "Title", "Text"])
+            output = gr.Dataframe(headers=["Score", "Title", "Text"])
         with gr.Column(scale=1):
             json = gr.JSON()
 
+    query.submit(search, inputs=[query], outputs=[output, json])
     search_button.click(search, inputs=[query], outputs=[output, json])
 
 demo.queue()