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

Update pages/04_Super_Resolution.py

Browse files
Files changed (1) hide show
  1. pages/04_Super_Resolution.py +5 -8
pages/04_Super_Resolution.py CHANGED
@@ -246,14 +246,11 @@ if restore:
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.button("Done")
252
- st.download_button(
253
- label="Download Image",
254
- data=open(output, "rb").read(),
255
- file_name="Image.jpg",
256
- )
257
  else:
258
  st.warning('Upload a file', icon="⚠️")
259
 
 
246
  output = realesrgan(input_img, model_name, denoise_strength,
247
  face_enhance, outscale)
248
  output_img.image(output, 'Restored Image')
249
+ st.download_button(
250
+ label="Download Image",
251
+ data=open(output, "rb").read(),
252
+ file_name="Image.jpg",
253
+ )
 
 
 
254
  else:
255
  st.warning('Upload a file', icon="⚠️")
256