Update app.py

app.py

@@ -1,14 +1,16 @@
+import os
+import cv2
+import gdown
 import gradio as gr
 import tensorflow as tf
-import gdown
 import urllib.request
-from PIL import Image
-
-import os
-import cv2
 import numpy as np
 import keras.backend as K
 
+from PIL import Image
+from matplotlib import pyplot as plt
+%matplotlib inline
+
 
 #from tensorflow import keras
 
@@ -71,7 +73,8 @@ def gen_pred(img=inputs, model=seg_model):
     print(pred)
     # color_coverted = cv2.cvtColor(pred, cv2.COLOR_BGR2RGB)
     # pil_image = Image.fromarray(color_coverted)
-    return "UI in developing process ..."
+    "UI in developing process ..."
+    return plt.imshow(pred, interpolation='catrom')
 
 
 title = "<h1 style='text-align: center;'>Semantic Segmentation</h1>"