add: downscale slider

app.py

@@ -51,6 +51,7 @@ def main():
   gr.Interface(m.pixelize_modified,
              [
               gr.components.Image(type='pil', label='Input'),
+              gr.components.Slider(minimum=1, maximum=8, value=1, step=1, label='Downscale Input'),
               gr.components.Slider(minimum=1, maximum=16, value=4, step=1, label='Pixel Size'),
               gr.components.Checkbox(True, label="Upscale after")
              ],

pixelization.py

@@ -41,14 +41,15 @@ class Model():
 
             save(out_t, out_img)
 
-    def pixelize_modified(self, in_img, pixel_size, upscale_after) -> Image.Image:
+    def pixelize_modified(self, in_img, downscale, pixel_size, upscale_after) -> Image.Image:
         with torch.no_grad():
             in_img = in_img.convert('RGB')
 
             # limit in_img size to 1024x1024 so it didn't destroyed by large image
             if in_img.size[0] > 1024 or in_img.size[1] > 1024:
                 in_img.thumbnail((1024, 1024), Image.NEAREST)
-
+            if downscale>1:
+              in_img.resize((in_img.size[0]//downscale, in_img.size[1]//downscale))
             in_img.resize((in_img.size[0] * 4 // pixel_size, in_img.size[1] * 4 // pixel_size))
 
             in_t = process(in_img).to(self.device)