Update app.py

app.py

@@ -82,8 +82,9 @@ with gr.Blocks() as demo:
             label="Generated image", 
             type='pil',
             show_label = False,
-            width="32",  # Use the full view width
-            height="32")  # Use the full view height
+            container = False,
+            width=32,  # Use the full view width
+            height=32)  # Use the full view height
     btn.click(gr_generate_images, [text], gallery)
     gr.Examples(examples, inputs=[text])
     gr.HTML(