Update pages/04_Super_Resolution.py

pages/04_Super_Resolution.py

@@ -249,11 +249,11 @@ if restore:
         #output_img.save("Image.jpg")
         with cols[1]:
             st.button("Done")
-        #    st.download_button(
-        #        label="Download Image",
-        #        data=open("Image.jpg", "rb").read(),
-        #        file_name="Image.jpg",
-        #    )
+            st.download_button(
+                label="Download Image",
+                data=open(output, "rb").read(),
+                file_name="Image.jpg",
+            )
     else:
         st.warning('Upload a file', icon="⚠️")