image-video-colorization

Runtime error

Cany-Tam commited on Sep 22, 2023

Commit

da86661

1 Parent(s): d6f7380

Update pages/04_Super_Resolution.py

Files changed (1) hide show

pages/04_Super_Resolution.py CHANGED Viewed

@@ -246,7 +246,7 @@ if restore:
         output = realesrgan(input_img, model_name, denoise_strength,
                             face_enhance, outscale)
         output_img.image(output, 'Restored Image')
-        st.download_button(
             label="Download Image",
             data=open(output, "rb").read(),
             file_name="Image.jpg",

         output = realesrgan(input_img, model_name, denoise_strength,
                             face_enhance, outscale)
         output_img.image(output, 'Restored Image')
+        output_img.download_button(
             label="Download Image",
             data=open(output, "rb").read(),
             file_name="Image.jpg",