Update app.py

app.py

@@ -21,12 +21,13 @@ def swap_gallery(im, images):
     return list(map(lambda x: x["name"], t2_list))
 
 
-def lexica(prompt, limit_size = 128):
+def lexica(prompt, limit_size = 128, resize = (512, 512)):
     lex = Lexica(query=prompt).images()
     lex = lex[:limit_size]
     lex = list(map(lambda ele: Image.open(
                     requests.get(ele, stream = True).raw
                     ), lex))
+    lex = list(map(lambda x: x.resize(resize), lex))
     return lex
 
 with gr.Blocks() as demo:
@@ -40,10 +41,10 @@ with gr.Blocks() as demo:
             outputs = gr.Gallery(lable='Output gallery').style(grid=3,height=768,
                 allow_preview=False)
             #gr.Dataframe(label='prompts for corresponding images')]
-            
-        #outputs.select(image_click, outputs, i, _js="(x) => x.splice(0,x.length)")    
+
+        #outputs.select(image_click, outputs, i, _js="(x) => x.splice(0,x.length)")
         outputs.select(image_click, outputs, i,)
-        
+
         i.change(
                 fn=swap_gallery,
                 inputs=[i, outputs],
@@ -54,4 +55,4 @@ with gr.Blocks() as demo:
 
         text_button.click(lexica, inputs=inputs, outputs=outputs)
 
-demo.launch("0.0.0.0" ,debug = True)
+demo.launch("0.0.0.0" ,debug = True)