Fabrice-TIERCELIN commited on
Commit
a81081c
1 Parent(s): e4f1721

gr.update(label="Downloadable results in *." + output_format + " format", format = output_format, value = [denoise_image])

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -250,7 +250,7 @@ def restore_in_Xmin(
250
 
251
  if torch.cuda.device_count() == 0:
252
  gr.Warning('Set this space to GPU config to make it work.')
253
- return [noisy_image, denoise_image], [denoise_image], None, gr.update(visible=True)
254
 
255
  if model_select != model.current_model:
256
  print('load ' + model_select)
 
250
 
251
  if torch.cuda.device_count() == 0:
252
  gr.Warning('Set this space to GPU config to make it work.')
253
+ return [noisy_image, denoise_image], gr.update(label="Downloadable results in *." + output_format + " format", format = output_format, value = [denoise_image]), None, gr.update(visible=True)
254
 
255
  if model_select != model.current_model:
256
  print('load ' + model_select)