Update app.py

app.py

@@ -17,17 +17,24 @@ def main():
         block = gr.Blocks()
         with block:
             gr.HTML('''
+                <pre>
                 This is an unconditioned diffusion model trained on around 500 
                 miscellaneous tiki images from around the web.
                 
                 It was trained for around 1300 epochs with a loss of 0.0249
                 or something... still running fix this later... uhm...
+                
+                It may take a really long time like 120s for some reason.
+                
+                The image size is hugely stretch atm... still trying to figure out 
+                how to control it...
+                </pre>
             ''')
             with gr.Group():
                 with gr.Box():
                     with gr.Row():
                         btn = gr.Button("Generate image")
-                        btn.click(imagine, inputs=None, outputs=gallery)
+                       
                 gallery = gr.Gallery(
                     label="Generated images", show_label=False, elem_id="gallery"
                 ).style(grid=[1], height="128px")