-

app.py

@@ -37,8 +37,8 @@ def seg(image):
     img = np.array(image) * 0.5 + color_seg * 0.5
     img = img.astype(np.uint8)
     img = Image.fromarray(img)
-    return plt.imshow(img)
+    return img
 
 
-iface = gr.Interface(fn=seg, inputs=gr.inputs.Image(type='pil'), outputs=gr.outputs.Image('filepath',True))
+iface = gr.Interface(fn=seg, inputs=gr.inputs.Image(type='pil'), outputs=gr.outputs.Image('pil'))
 iface.launch()