Update app.py

app.py

@@ -8,7 +8,7 @@ import numpy as np
 import keras.backend as K
 
 from PIL import Image
-
+from matplotlib import cm
 
 #from tensorflow import keras
 
@@ -79,7 +79,7 @@ def gen_pred(img=inputs, model=seg_model):
     pred = model.predict(img)
     pred = np.squeeze(pred, axis=0)
     print(pred)
-    pred = fig2img(pred)
+    pred = Image.fromarray(np.uint8(cm.gist_earth(pred)*255))
     # color_coverted = cv2.cvtColor(pred, cv2.COLOR_BGR2RGB)
     # pil_image = Image.fromarray(pred)
     # PIL_image = Image.fromarray(pred.astype('uint8'), 'RGB')