added Gallery style

app.py

@@ -25,20 +25,14 @@ def url2img(url, resize = False, fix_height = 200):
     data = requests.get(url, allow_redirects = True).content
     img = Image.open(io.BytesIO(data))
     if resize:
-        width, height = img.size
-        img = img.resize((resize_wh(width, height, fix_height)))
+         img.thumbnail([fix_height, fix_height], Image.LANCZOS)
     return  img
 
-def resize_wh(w, h, fix_height):
-    ratio = w / h
-    w_n = int(fix_height * ratio)
-    return w_n, fix_height
-
 def find_topk(text):
 
     print('text', text)
 
-    top_k = 10
+    top_k = 20
 
     text_data = model_multi.preprocess_text(text)
     text_features, text_embedding = model_multi.encode_text(text_data, return_features=True)
@@ -53,7 +47,7 @@ def find_topk(text):
     print('top_k_urls', top_k_urls)
     print(datetime.now().strftime("%H:%M:%S"))
 
-    images = [url2img(url, resize = True) for url in top_k_urls]
+    images = [url2img(url) for url in top_k_urls]
 
     print('got PIL images')
     print(datetime.now().strftime("%H:%M:%S"))
@@ -97,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()
+        gallery = gr.Gallery().style(columns=[5], rows=[4], object_fit="contain", height="auto")
         btn_search.click(find_topk, inputs = prompt_box, outputs = gallery)
 
 if __name__ == "__main__":