Update app.py

app.py

@@ -1,11 +1,19 @@
 import gradio as gr
 from main import main
 
-iface = gr.Interface(
-        fn=main,
-        inputs=gr.Image(type="filepath", label="Invoice Image"),
-        outputs=gr.Image(type="pil", label="Masked Invoice Image"),
-        live=True
-    )
-
-iface.launch()
+
+def main(image):
+    return image
+
+with gr.Blocks(title="Privacy Mask", css="footer{display:none !important}") as demo:
+    
+    with gr.Row():
+        image_input = gr.Image(type="filepath", label="Invoice Image")
+        image_output = gr.Image(show_download_button=True, type="pil", label="Masked Invoice Image")
+    
+    with gr.Row():
+        image_button = gr.Button("Masking")
+        image_button.click(main, inputs=image_input, outputs=image_output)
+        clear_button = gr.ClearButton([image_input, image_output])
+
+demo.launch()