Update app.py
Browse files
app.py
CHANGED
@@ -165,7 +165,7 @@ def inference(image, upscale, large_input_flag, color_fix):
|
|
165 |
output = np.transpose(output[[2, 1, 0], :, :], (1, 2, 0))
|
166 |
output = (output * 255.0).round().astype(np.uint8)
|
167 |
|
168 |
-
return
|
169 |
|
170 |
# # save restored img
|
171 |
# save_path = f'results/out.png'
|
|
|
165 |
output = np.transpose(output[[2, 1, 0], :, :], (1, 2, 0))
|
166 |
output = (output * 255.0).round().astype(np.uint8)
|
167 |
|
168 |
+
return img, Image.fromarray(output)
|
169 |
|
170 |
# # save restored img
|
171 |
# save_path = f'results/out.png'
|