Update app.py

app.py

@@ -40,18 +40,47 @@ def generate():
             noise_pred = image_pipe.unet(model_input, t)["sample"]
         x = scheduler.step(noise_pred, t, x).prev_sample
     # View the results
-    return show_images_save(x).resize((64, 64), resample=Image.NEAREST)
-    
+    return show_images_save(x)
 
-outputs = (gr.Image(label="result"), gr.Image(label="result"))
+def ex():
+  return generate(), generate(), generate(), generate()
 
-# Setting up a minimal interface to our function:
-demo = gr.Interface(
-    fn=generate,
-    inputs=None,
-    outputs=outputs,
-)
+demo = gr.Blocks()
+
+with demo:
+  gr.HTML(
+        """
+            <div style="text-align: center; margin: 0 auto;">
+              <div style="display: inline-flex;align-items: center;gap: 0.8rem;font-size: 1.75rem;">
+                <h1 style="font-weight: 900; margin-bottom: 7px;margin-top:5px">
+                  Minecraft Skin Diffusion
+                </h1>
+              </div>
+              <p style="margin-bottom: 10px; font-size: 94%; line-height: 23px;">
+                Gradio demo for Minecraft Skin Diffusion. This is simple Unconditional Diffusion Model that will help you generate skins for game Minecraft.
+              </p>
+            </div>
+        """
+    )
+  with gr.Column(scale=4):
+    with gr.Row():
+      out = gr.Image(shape=(64,64), image_mode='RGBA', type='pil')
+      out2 = gr.Image(shape=(64,64), image_mode='RGBA', type='pil')
+    with gr.Row():
+      out3 = gr.Image(shape=(64,64), image_mode='RGBA', type='pil')
+      out4 = gr.Image(shape=(64,64), image_mode='RGBA', type='pil')
+  greet_btn = gr.Button("Generate")
+  greet_btn.click(fn=ex, inputs=None, outputs=[out, out2, out3, out4])
+  gr.HTML(
+            """
+                <div class="footer">
+                    <div style='text-align: center;'>Minecraft Skin Diffusion by <a href='https://twitter.com/wine_ineff' target='_blank'>Artsem Holub (WiNE-iNEFF)</a> | 
+                      <center>
+                        <img src='https://visitor-badge.glitch.me/badge?page_id=WiNE-iNEFF_MinecraftSkin-Diffusion' alt='visitor badge'>
+                      </center>
+                    </div>
+               </div>
+           """
+        )
 
-# And launching
-if __name__ == "__main__":
-    demo.launch(enable_queue=True)
+demo.launch()