ledetele commited on
Commit
e6e25be
·
1 Parent(s): 812dd64

Update pages/05_Image_Denoizer.py

Browse files
Files changed (1) hide show
  1. pages/05_Image_Denoizer.py +6 -0
pages/05_Image_Denoizer.py CHANGED
@@ -241,6 +241,12 @@ if restore:
241
  output = realesrgan(input_img, model_name, denoise_strength,
242
  face_enhance, outscale)
243
  output_img.image(output, 'Restored Image')
 
 
 
 
 
 
244
  else:
245
  st.warning('Upload a file', icon="⚠️")
246
 
 
241
  output = realesrgan(input_img, model_name, denoise_strength,
242
  face_enhance, outscale)
243
  output_img.image(output, 'Restored Image')
244
+
245
+ st.download_button(
246
+ label="Download Image",
247
+ data=open(output, "rb").read(),
248
+ file_name="Image.jpg",
249
+ )
250
  else:
251
  st.warning('Upload a file', icon="⚠️")
252