Update app.py

app.py

@@ -2,8 +2,8 @@ import gradio as gr
 
 
 demo = gr.Interface(
-    inputs=[gr.Image()], 
-    outputs=inputs,
+    inputs=[inputs], 
+    outputs="image",
     title="MaskFormer Demo"
 )