Fix by rollback
Browse files
app.py
CHANGED
|
@@ -208,7 +208,6 @@ def restore_in_Xmin(
|
|
| 208 |
return [noisy_image, denoise_image], [denoise_image], None, None
|
| 209 |
|
| 210 |
denoise_image = HWC3(np.array(denoise_image))
|
| 211 |
-
denoise_image = upscale_image(denoise_image, upscale, unit_resolution=32, min_size=min_size)
|
| 212 |
|
| 213 |
# Allocation
|
| 214 |
if allocation == 1:
|
|
@@ -335,6 +334,7 @@ def restore(
|
|
| 335 |
model.load_state_dict(ckpt_F, strict=False)
|
| 336 |
model.current_model = model_select
|
| 337 |
|
|
|
|
| 338 |
LQ = np.array(input_image) / 255.0
|
| 339 |
LQ = np.power(LQ, gamma_correction)
|
| 340 |
LQ *= 255.0
|
|
|
|
| 208 |
return [noisy_image, denoise_image], [denoise_image], None, None
|
| 209 |
|
| 210 |
denoise_image = HWC3(np.array(denoise_image))
|
|
|
|
| 211 |
|
| 212 |
# Allocation
|
| 213 |
if allocation == 1:
|
|
|
|
| 334 |
model.load_state_dict(ckpt_F, strict=False)
|
| 335 |
model.current_model = model_select
|
| 336 |
|
| 337 |
+
input_image = upscale_image(input_image, upscale, unit_resolution=32, min_size=min_size)
|
| 338 |
LQ = np.array(input_image) / 255.0
|
| 339 |
LQ = np.power(LQ, gamma_correction)
|
| 340 |
LQ *= 255.0
|