Fabrice-TIERCELIN commited on
Commit
11fe5b3
1 Parent(s): 7c4ec71

Fix function

Browse files
Files changed (1) hide show
  1. gradio_demo.py +1 -1
gradio_demo.py CHANGED
@@ -146,7 +146,7 @@ def stage2_process(
146
  model_select,
147
  output_format
148
  ):
149
- restore(
150
  noisy_image,
151
  denoise_image,
152
  prompt,
 
146
  model_select,
147
  output_format
148
  ):
149
+ return restore(
150
  noisy_image,
151
  denoise_image,
152
  prompt,