Fix function
Browse files- 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,
|