Update pages/04_Super_Resolution.py

pages/04_Super_Resolution.py

@@ -246,7 +246,13 @@ if restore:
         output = realesrgan(input_img, model_name, denoise_strength, 
                             face_enhance, outscale)
         output_img.image(output, 'Restored Image')
+        output_img.save("Image.jpg")
         with cols[1]:
+            st.download_button(
+                label="Download Image",
+                data=open("Image.jpg", "rb").read(),
+                file_name="Image.jpg",
+            )
             out_filename = "enlarged_images.zip"
             # Provide the zip file data for download
             st.download_button(