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
|