After upscaling an image, there is no option to download the upscaled version. When I click the Download button, only the original image is downloaded instead of the upscaled image.

After upscaling an image, there is no option to download the upscaled version. When I click the Download button, only the original image is downloaded instead of the upscaled image.

Steps to Reproduce:

1. Upload an image.
2. Click Upscale.
3. Wait for the upscaling process to complete.
4. Click Download.

Expected Result:
The downloaded file should be the upscaled image.

Actual Result:
Only the original image is downloaded, and there is no option to download the upscaled version.

Additional Information:
The issue occurs after the upscaling process completes successfully.