-

app.py

@@ -3,7 +3,7 @@ from torch import nn
 import numpy as np
 import matplotlib.pyplot as plt
 from palette import ade_palette
-
+from PIL import Image
 import gradio as gr
 
 def seg(image):
@@ -36,7 +36,7 @@ def seg(image):
     color_seg = color_seg[..., ::-1]
     img = np.array(image) * 0.5 + color_seg * 0.5
     img = img.astype(np.uint8)
-
+    img = Image.fromarray(img)
     return plt.imshow(img)