Update app.py

app.py

@@ -300,8 +300,8 @@ if __name__ == '__main__':
 
             pred_mask = decode_from_colormap(pred_mask, colors)[..., ::-1]
 
-            return pred_mask.astype(np.uint8)
-
+            return Image.fromarray(pred_mask.astype(np.uint8))
+             
 
     demo = gr.Interface(
         fn=predict_image,