Update app.py

app.py

@@ -25,7 +25,14 @@ def show_images_save(x):
     grid = torchvision.utils.make_grid(x, nrow=4, padding=0)
     grid_im = grid.detach().cpu().permute(1, 2, 0).clip(0, 1) * 255
     grid_im = Image.fromarray(np.array(grid_im).astype(np.uint8))
-    return grid_im
+    #CROP
+    width, height = grid_im.size
+    sav = []
+    for w in range(width//64):
+      for h in range(height//64):
+        box = (w*64, h*64, (w+1)*64, (h+1)*64)
+        sav.append(grid_im.crop(box))
+    return sav
 
 def color_loss(images, target_color=(0, 0, 0, 1)):
     target = (torch.tensor(target_color).to(images.device) * 2 - 1)
@@ -34,6 +41,7 @@ def color_loss(images, target_color=(0, 0, 0, 1)):
     return error
 
 def generate():
+    print(ctime(time()))
     x = torch.randn(8, 4, 64, 64).to(device)
     for i, t in enumerate(scheduler.timesteps):
         model_input = scheduler.scale_model_input(x, t)
@@ -43,7 +51,7 @@ def generate():
     return show_images_save(x)
 
 def generate_g(color, guidance_loss_scale):
-    print('--V2--')
+    print(f'--V2-- {ctime(time())}')
     target_color = ImageColor.getcolor(color, "RGBA")  # Target color as RGB
     target_color = [a / 255 for a in target_color]  # Rescale from (0, 255) to (0, 1)
     x = torch.randn(8, 4, 64, 64).to(device)
@@ -59,54 +67,34 @@ def generate_g(color, guidance_loss_scale):
         x = scheduler.step(noise_pred, t, x).prev_sample
     return show_images_save(x)
 
-def crrop(file):
-    width, height = file.size
-    sav = []
-    for w in range(width//64):
-      for h in range(height//64):
-        box = (w*64, h*64, (w+1)*64, (h+1)*64)
-        sav.append(file.crop(box))
-    return sav
-
-def ex():
-    t = time()
-    print(ctime(t))
-    return crrop(generate())
-
-def ex_g(picker, slider):
-    t = time()
-    print(ctime(t))
-    return crrop(generate_g(picker, slider))
-
-demo = gr.Blocks(css=".container {max-width: 730px;margin: auto;} #gallery {margin-bottom: 15px; margin-left: auto; margin-right: auto; border-bottom-right-radius: .5rem !important; border-bottom-left-radius: .5rem !important;}")
-
+demo = gr.Blocks(css=".tabs {max-width: 60%; margin-left: 20%; margin-right: 20%;} #column {max-width: 50%} .fixed-height {min-height: 0px !important} @media(min-width: 1280px){.fixed-height {min-height: 0px !important}}")
 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;">
-                <img src='https://huggingface.co/spaces/WiNE-iNEFF/MinecraftSkin-Diffusion/resolve/main/MSD_7.png'>
+            <div style="display: inline-flex; align-items: center; justify-content: center;">
+                <img src='https://huggingface.co/spaces/WiNE-iNEFF/MinecraftSkin-Diffusion/resolve/main/MSD_7.png' width='45%'>
             </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.
+                Gradio demo for Minecraft Skin Diffusion. This is simple Unconditional Diffusion Model<br>that can generate skins for game Minecraft.
             </p>
         </div>
         """)
     with gr.Tabs():
         with gr.TabItem("V1"):
             with gr.Column():
-                with gr.Row().style(equal_height=True):
-                    gall = gr.Gallery(elem_id='gallery').style(grid=[4])
+                gall = gr.Gallery().style(grid=[4])
                 greet_btn = gr.Button("Generate")
-                greet_btn.click(fn=ex, outputs=gall)
+                greet_btn.click(fn=generate, outputs=gall)
         with gr.TabItem("V2"):
-            with gr.Column():
-                with gr.Row().style(equal_height=True):
-                    picker = gr.ColorPicker(label="color", value="#55FFAA")
+            with gr.Row():
+                with gr.Column(min_width = 0, elem_id='column'):
+                    picker = gr.ColorPicker(label="color", value="#55FFAA", min_width='0')
                     slide = gr.Slider(label="guidance_scale", minimum=0, maximum=100, value=50)
-                gall = gr.Gallery(elem_id='gallery').style(grid=[4])
-                greet_btn = gr.Button("Generate")
-                greet_btn.click(fn=ex_g, inputs=[picker, slide], outputs=gall)
+                    greet_btn = gr.Button("Generate", min_width = '0')
+                with gr.Column(min_width = 0):
+                  gall = gr.Gallery(min_width = '250').style(grid=[3])
+                greet_btn.click(fn=generate_g, inputs=[picker, slide], outputs=gall)
     gr.HTML(
         """
         <div class="footer">