Update app.py

app.py

@@ -262,4 +262,19 @@ if (window.gradio_config.version.startsWith('3')) {
 """
 
 iface = gr.Interface(
-    fn=remove_background
+    fn=remove_background,
+    inputs=gr.Image(type="numpy"),
+    outputs=[
+        gr.Image(type="filepath", label="Mask"),
+        gr.Image(type="filepath", label="Reverse Mask"),
+        gr.Image(type="filepath", label="Foreground"),
+        gr.Image(type="filepath", label="Background")
+    ],
+    allow_flagging="never",
+    css=css,
+    js=js,
+    elem_id="remove-background"
+)
+
+if __name__ == "__main__":
+    iface.launch(debug=True)