Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -27,7 +27,7 @@ def inference(file, mode):
|
|
27 |
|
28 |
if mode == "enhance":
|
29 |
return os.path.join("e.png")
|
30 |
-
|
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
|