Update app.py

app.py

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