Update app.py

app.py

@@ -68,9 +68,10 @@ def gen_pred(img=inputs, model=seg_model):
     img = tf.expand_dims(img, axis=0)
     pred = model.predict(img)
     pred = np.squeeze(pred, axis=0)
+    print(pred)
     # color_coverted = cv2.cvtColor(pred, cv2.COLOR_BGR2RGB)
     # pil_image = Image.fromarray(color_coverted)
-    return pred
+    return "UI in developing process ..."
 
 
 title = "<h1 style='text-align: center;'>Semantic Segmentation</h1>"