plot

app.py

@@ -1,6 +1,7 @@
 from transformers import AutoFeatureExtractor, SegformerForSemanticSegmentation
 from torch import nn
 import numpy as np
+import matplotlib.pyplot as plt
 from palette import ade_palette
 
 import gradio as gr
@@ -36,7 +37,8 @@ def seg(image):
     img = np.array(image) * 0.5 + color_seg * 0.5
     img = img.astype(np.uint8)
 
-    return img
+    return plt.imshow(img)
+
 
 iface = gr.Interface(fn=seg, inputs=gr.inputs.Image(type='pil'), outputs=gr.outputs.Image('numpy',True))
 iface.launch()