Update app.py

app.py

@@ -44,6 +44,7 @@ def generate(
     num_inference_steps_base: int = 25,
     num_inference_steps_refiner: int = 25,
     apply_refiner: bool = False,
+    pixel_size: int = 16
 ):
     client = Client("hysts/SDXL", hf_token=os.environ['HF_TOKEN'])
     result = client.predict(
@@ -64,7 +65,7 @@ def generate(
         apply_refiner=apply_refiner,
         api_name="/run",
     )
-    image = pixelate(result, 16)
+    image = pixelate(result, pixel_size)
     return image
 
 
@@ -169,6 +170,13 @@ with gr.Blocks(css="style.css") as demo:
                 step=1,
                 value=25,
             )
+        pixel_size = gr.Slider(
+            label="Pixel size",
+            minimum=1,
+            maximum=64,
+            step=1,
+            value=16,
+        )
 
     gr.Examples(
         examples=examples,
@@ -237,6 +245,7 @@ with gr.Blocks(css="style.css") as demo:
             num_inference_steps_base,
             num_inference_steps_refiner,
             apply_refiner,
+            pixel_size
         ],
         outputs=result,
         api_name="run",