Update app.py

app.py

@@ -8,7 +8,6 @@ import numpy as np
 import keras.backend as K
 
 from PIL import Image
-from matplotlib import pyplot as plt
 
 
 #from tensorflow import keras
@@ -71,9 +70,9 @@ def gen_pred(img=inputs, model=seg_model):
     pred = np.squeeze(pred, axis=0)
     print(pred)
     # color_coverted = cv2.cvtColor(pred, cv2.COLOR_BGR2RGB)
-    # pil_image = Image.fromarray(color_coverted)
+    pil_image = Image.fromarray(pred)
     "UI in developing process ..."
-    return plt.imshow(pred, interpolation='catrom')
+    return pil_image
 
 
 title = "<h1 style='text-align: center;'>Semantic Segmentation</h1>"