Fabrice-TIERCELIN commited on
Commit
ff6a8f8
1 Parent(s): 77c3bd5

Fix by rollback

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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