yuragoithf commited on
Commit
2d24604
·
1 Parent(s): 88d04c7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -2
app.py CHANGED
@@ -64,8 +64,6 @@ def gen_pred(img, model=seg_model):
64
  img = tf.expand_dims(img, axis=0)
65
  pred = model.predict(img)
66
  pred = np.squeeze(pred, axis=0)
67
- print(len(pred))
68
- print(max(pred))
69
  img_bytes = pred.tobytes()
70
  nparr = np.frombuffer(img_bytes, np.byte)
71
  pred_pil = cv2.imdecode(nparr, cv2.IMREAD_REDUCED_COLOR_8)
 
64
  img = tf.expand_dims(img, axis=0)
65
  pred = model.predict(img)
66
  pred = np.squeeze(pred, axis=0)
 
 
67
  img_bytes = pred.tobytes()
68
  nparr = np.frombuffer(img_bytes, np.byte)
69
  pred_pil = cv2.imdecode(nparr, cv2.IMREAD_REDUCED_COLOR_8)