Cany-Tam commited on
Commit
ae58611
·
1 Parent(s): b9859fb

Update pages/04_Super_Resolution.py

Browse files
Files changed (1) hide show
  1. pages/04_Super_Resolution.py +6 -0
pages/04_Super_Resolution.py CHANGED
@@ -246,7 +246,13 @@ if restore:
246
  output = realesrgan(input_img, model_name, denoise_strength,
247
  face_enhance, outscale)
248
  output_img.image(output, 'Restored Image')
 
249
  with cols[1]:
 
 
 
 
 
250
  out_filename = "enlarged_images.zip"
251
  # Provide the zip file data for download
252
  st.download_button(
 
246
  output = realesrgan(input_img, model_name, denoise_strength,
247
  face_enhance, outscale)
248
  output_img.image(output, 'Restored Image')
249
+ output_img.save("Image.jpg")
250
  with cols[1]:
251
+ st.download_button(
252
+ label="Download Image",
253
+ data=open("Image.jpg", "rb").read(),
254
+ file_name="Image.jpg",
255
+ )
256
  out_filename = "enlarged_images.zip"
257
  # Provide the zip file data for download
258
  st.download_button(