Update app.py

app.py

@@ -31,10 +31,11 @@ theme = gr.themes.Base(
     input_background_fill='#ffffff',
     #input_border_width='*block_border_width',
     button_primary_background_fill='#ffffff',
+    button_border_widt='*block_border_width',
 )
 
 with gr.Blocks(theme=theme) as demo:
-    img = gr.Image(show_label=False, type='pil', height=512, width=512)
+    img = gr.Image(show_label=False, type='pil')
     textbox = gr.Textbox(show_label=False, placeholder='type your prompt in here')
     button = gr.Button("generate",  variant="primary")
     button.click(fn=generate, inputs=textbox, outputs=img)