Update pages/04_Super_Resolution.py

pages/04_Super_Resolution.py

@@ -246,6 +246,14 @@ if restore:
         output = realesrgan(input_img, model_name, denoise_strength, 
                             face_enhance, outscale)
         output_img.image(output, 'Restored Image')
+        with cols[1]:
+            out_filename = "enlarged_images.zip"
+            # Provide the zip file data for download
+            st.download_button(
+                label= "Download Colorized Image",
+                data=open(out_filename, "rb").read(),
+                file_name=out_filename,
+            )        
     else:
         st.warning('Upload a file', icon="⚠️")