smaller Gallery grid

app.py

@@ -32,7 +32,7 @@ def find_topk(text):
 
     print('text', text)
 
-    top_k = 20
+    top_k = 10
 
     text_data = model_multi.preprocess_text(text)
     text_features, text_embedding = model_multi.encode_text(text_data, return_features=True)
@@ -91,7 +91,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                 prompt_box = gr.Textbox(label = 'Enter your prompt', lines = 3)
                 btn_search = gr.Button("Find images")
 
-        gallery = gr.Gallery().style(columns=5, rows=4, object_fit="contain", height="auto")
+        gallery = gr.Gallery().style(columns=5, rows=2, object_fit="contain", height="auto")
         btn_search.click(find_topk, inputs = prompt_box, outputs = gallery)
 
 if __name__ == "__main__":