Update app.py

app.py

@@ -16,7 +16,7 @@ def convert(input_img, quality=85):
     
     return img_base64
 
-def send_req(input_img: PIL.Image.Image):
+def send_req(input_img, compression, noise):
 
     if type(input_img) is not PIL.Image.Image:
         input_img = Image.fromarray(input_img, 'RGB')
@@ -24,7 +24,9 @@ def send_req(input_img: PIL.Image.Image):
     payload = {
       "input": {
         "image": convert(input_img),
-        "mode": "1"
+        "mode": "1",
+        "quality": str(compression),
+        "noise": str(noise)
       }
     }
 
@@ -40,6 +42,6 @@ def send_req(input_img: PIL.Image.Image):
     
     return image
 
-demo = gr.Interface(send_req, gr.Image(), "image")
+demo = gr.Interface(send_req, [gr.Image(), gr.slider(0, 100, label="Compression", step=1), gr.slider(0, 100, label="Noise", step=1)], "image")
 if __name__ == "__main__":
     demo.launch()