add n to quantize

app.py

@@ -2,15 +2,15 @@ import numpy as np
 
 import gradio as gr
 
-def quantize(val):
-    return (np.round(val / 255) * 255).astype(np.uint8)
+def quantize(val, n):
+    return (np.round(n * val / 255) * 255).astype(np.uint8)
 
 
-def sepia(input_img):
-    output_img = list(map(quantize, input_img))
+def sepia(n, input_img):
+    output_img = [quantize(val, n) for val in input_img]
     return output_img
 
 
-iface = gr.Interface(sepia, "image", "pil")
+iface = gr.Interface(sepia, ["number", "image"], "pil")
 
 iface.launch()