Update app.py

app.py

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