finish

app.py

@@ -105,11 +105,11 @@ def sepia(input_img):
     return fig
 
 
-demo = gr.Interface(fn=sepia,
-                    inputs=gr.Image(shape=(1024, 1024)),
-                    outputs=['plot'],
+iface = gr.Interface(fn=sepia,
+                    inputs=gr.inputs.Image(type="pil", label="Upload an image"),
+                    outputs=gr.outputs.Image(type='pil', label="Segmentation result"),
                     examples=["city-1.jpg", "city-2.jpg", "city-3.jpg", "city-4.jpg", "city-5.jpg"],
                     allow_flagging='never')
 
 
-demo.launch()
+iface.launch()