Update app.py

app.py

@@ -1,10 +1,11 @@
-import os
+import os, io
 import cv2
 import gradio as gr
 import tensorflow as tf
 import urllib.request
 import numpy as np
 import keras.backend as K
+from transformers import pipeline
 
 from PIL import Image
 from matplotlib import cm
@@ -56,17 +57,20 @@ def gen_pred(img, model=seg_model):
     # open_cv_image = np.array(pil_image)
     # img = open_cv_image[:, :, ::-1].copy() 
     # # img = cv2.imread("./003e2c95d.jpg")
+    img_byte_arr = io.BytesIO()
+    img.save(img_byte_arr, format='JPEG', subsampling=0, quality=100)
+    img = img_byte_arr.getvalue()
     img = img[::IMG_SCALING[0], ::IMG_SCALING[1]]
     img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
     img = img/255
     img = tf.expand_dims(img, axis=0)
     pred = model.predict(img)
     pred = np.squeeze(pred, axis=0)
-    print(pred)
+    print(len(pred))
+    print(max(pred))
     img_bytes = pred.tobytes()
     nparr = np.frombuffer(img_bytes, np.byte)
     pred_pil = cv2.imdecode(nparr, cv2.IMREAD_REDUCED_COLOR_8)
-    print(pred_pil)
     # return "UI in developing process ..."
     return pred_pil