KenjieDec commited on
Commit
3c57a21
·
1 Parent(s): c9f5c91

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ def inference(file, mode):
27
 
28
  if mode == "enhance":
29
  return os.path.join("e.png")
30
- else if mode == "colorize":
31
  model = {'name':'GPEN-1024-Color', 'size':1024}
32
  grayf = cv2.imread("e.png", cv2.IMREAD_GRAYSCALE)
33
  grayf = cv2.cvtColor(grayf, cv2.COLOR_GRAY2BGR) # channel: 1->3
 
27
 
28
  if mode == "enhance":
29
  return os.path.join("e.png")
30
+ elif mode == "colorize":
31
  model = {'name':'GPEN-1024-Color', 'size':1024}
32
  grayf = cv2.imread("e.png", cv2.IMREAD_GRAYSCALE)
33
  grayf = cv2.cvtColor(grayf, cv2.COLOR_GRAY2BGR) # channel: 1->3