Update app.py
Browse files
app.py
CHANGED
|
@@ -154,7 +154,7 @@ def inference(image, upscale, large_input_flag, color_fix):
|
|
| 154 |
|
| 155 |
# save results
|
| 156 |
save_path = './out.png'
|
| 157 |
-
cv2.imwrite(save_path, output)
|
| 158 |
|
| 159 |
return (image, Image.fromarray(output)), save_path
|
| 160 |
|
|
|
|
| 154 |
|
| 155 |
# save results
|
| 156 |
save_path = './out.png'
|
| 157 |
+
cv2.imwrite(save_path, output[:, :, ::-1])
|
| 158 |
|
| 159 |
return (image, Image.fromarray(output)), save_path
|
| 160 |
|