Update pages/04_图像放大.py

pages/04_图像放大.py

@@ -246,6 +246,13 @@ if restore:
         output = realesrgan(input_img, model_name, denoise_strength, 
                             face_enhance, outscale)
         output_img.image(output, '恢复图像')
+
+        st.download_button(
+            label="下载图片",
+            data=open(output, "rb").read(),
+            file_name="Image.jpg",
+        )
+        
     else:
         st.warning('请上传文件', icon="⚠️")