Update app.py

app.py

@@ -30,7 +30,9 @@ with col1:
             
             st.subheader('New image')
             st.image(img, caption=f'Upscaled {choice} times image')
-            st.download_button(label="Download image", data=img, file_name="upscaled.png", mime="image/png")
+
+            with open('./scaled_2x.png', 'rb') as f:
+                st.download_button(label="Download image", data=f, file_name="upscaled.png", mime="image/png")
             
         else:
             st.text('Not supported yet. Comeback soon for fun releases')