Update pages/04_Super_Resolution.py

pages/04_Super_Resolution.py

@@ -248,18 +248,12 @@ if restore:
         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",
+            st.button("Done")
+        #    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(
-                label= "Download Colorized Image",
-                data=open(out_filename, "rb").read(),
-                file_name=out_filename,
-            )        
     else:
         st.warning('Upload a file', icon="⚠️")