Update app.py

app.py

@@ -19,28 +19,15 @@ device = (
 
 pipeline_name = 'WiNE-iNEFF/Minecraft-Skin-Diffusion-V2'
 image_pipe = DDPMPipeline.from_pretrained(pipeline_name).to(device)
-'''
-class Model:
-    def __init__(self, name):
-        self.name = name
-model = [
-    Model("DDIMScheduler"),
-    Model("PNDMScheduler")]
-current_model = model[0]
-'''
+
 def show_images_save(x):
     grid = torchvision.utils.make_grid(x, nrow=4)
     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
 
+
 def generate():
-    '''
-    if schedul == "DDIMScheduler":
-        scheduler = DDIMScheduler.from_pretrained(pipeline_name)
-    else:
-        scheduler = PNDMScheduler.from_pretrained(pipeline_name)
-    '''
     scheduler = DDIMScheduler.from_pretrained(pipeline_name)
     scheduler.set_timesteps(num_inference_steps=20)
     x = torch.randn(8, 4, 64, 64).to(device)
@@ -51,6 +38,7 @@ def generate():
         x = scheduler.step(noise_pred, t, x).prev_sample
     return show_images_save(x)
 
+
 def crrop(file):
     width, height = file.size
     sav = []
@@ -60,47 +48,38 @@ def crrop(file):
         sav.append(file.crop(box))
     return sav
 
+
 def ex():
     t = time()
     print(ctime(t))
     return crrop(generate())
 
-demo = gr.Blocks(css="#gallery {max-height: 256px} .container {max-width: 730px; margin: auto;} .min-h-\[15rem\]{min-height: 5rem !important;}")
+
+demo = gr.Blocks(css=".container {max-width: 730px; margin: auto;} .min-h-\[15rem\]{min-height: 5rem !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>
-              <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>
-        """
+    """
+    <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>
+        <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():
-    '''
-    with gr.Row().style(equal_height=True):
-        model_name = gr.Dropdown(label="Base Scheduler", choices=[m.name for m in model], value=current_model.name)
-        #number = gr.Number(value="40", label="number of generation steps (Standard value 40, MAX 1000; The larger the number, the better the quality, but the longer it takes)", show_label=True)
-    '''
     with gr.Row().style(equal_height=True):
-      gall = gr.Gallery(elem_id='gallery')
-  greet_btn = gr.Button("Generate")
-  greet_btn.click(fn=ex, outputs=gall)
-  '''inputs=[model_name]'''
+        gall = gr.Gallery(elem_id='gallery')
+    greet_btn = gr.Button("Generate")
+    greet_btn.click(fn=ex, outputs=gall)
   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>
-           """
-        )
+    """
+    <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></div>
+    </div>
+    """)
 
 demo.launch()