Update app.py

app.py

@@ -2,12 +2,12 @@ import gradio as gr
 from skimage import io
 from pyxelate import Pyx, Pal
 
-def pixel(image):
+def pixel(image,downsample,palette,depth,upscale):
     image = io.imread(image.name)  
-    downsample_by = 14  # new image will be 1/14th of the original in size
-    palette = 7  # find 7 colors
+    downsample_by = downsample  # new image will be 1/14th of the original in size
+    palette = palette  # find 7 colors
     # 1) Instantiate Pyx transformer
-    pyx = Pyx(factor=downsample_by, palette=palette,depth=2,upscale = 14)
+    pyx = Pyx(factor=downsample_by, palette=palette,depth=depth,upscale = upscale)
     # 2) fit an image, allow Pyxelate to learn the color palette
     pyx.fit(image)
     # 3) transform image to pixel art using the learned color palette
@@ -22,7 +22,8 @@ article = "<p style='text-align: center'><a href='https://github.com/sedthh/pyxe
 
 gr.Interface(
     pixel, 
-    [gr.inputs.Image(type="file", label="Input")], 
+    [gr.inputs.Image(type="file", label="Input"),gr.inputs.Number(default=14, label="downsample by"),gr.inputs.Number(default=7, label="palette"),gr.inputs.Number(default=2, label="depth"),gr.inputs.Number(default=14, label="upscale")
+], 
     gr.outputs.Image(type="file", label="Output"),
     title=title,
     description=description,