yuragoithf commited on
Commit
3821d52
·
1 Parent(s): 8a7486c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -53,10 +53,10 @@ image_output = gr.outputs.Image(type="pil", label="Output Image")
53
 
54
 
55
  def gen_pred(img, model=seg_model):
56
- pil_image = img.convert('RGB')
57
- open_cv_image = np.array(pil_image)
58
- img = open_cv_image[:, :, ::-1].copy()
59
- # img = cv2.imread("./003e2c95d.jpg")
60
  img = img[::IMG_SCALING[0], ::IMG_SCALING[1]]
61
  img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
62
  img = img/255
 
53
 
54
 
55
  def gen_pred(img, model=seg_model):
56
+ # pil_image = img.convert('RGB')
57
+ # open_cv_image = np.array(pil_image)
58
+ # img = open_cv_image[:, :, ::-1].copy()
59
+ # # img = cv2.imread("./003e2c95d.jpg")
60
  img = img[::IMG_SCALING[0], ::IMG_SCALING[1]]
61
  img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
62
  img = img/255