yuragoithf commited on
Commit
363ec41
·
1 Parent(s): d1eaab8

Update app.py

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