Update pages/05_Image_Denoizer.py

pages/05_Image_Denoizer.py

@@ -214,22 +214,23 @@ outscale = 1
 face_enhance = False
 
 if input_img:
-  print(input_img)
+  #print(input_img)
   input_img = Image.open(input_img)
   # Display image properties
-  cols = st.columns(2)
+  col1, col2 = st.columns(2)
 
-  cols[0].image(input_img, 'Source Image')
+  with col1:
+    st.image(input_img, 'Source Image')
 
   #input_properties = get_image_properties(input_img)
   #cols[1].write(input_properties)
 
   # Output placeholder
-  output_img = st.empty()
+  #output_img = st.empty()
 
 # Input and output placeholders
-input_img = input_img
-output_img = st.empty()
+#input_img = input_img
+#output_img = st.empty()
 
 # Buttons
 restore = st.button('Restore')
@@ -240,13 +241,16 @@ if restore:
     if input_img is not None:
         output = realesrgan(input_img, model_name, denoise_strength, 
                             face_enhance, outscale)
-        output_img.image(output, 'Restored Image')
+        with col2:
+          st.image(output, 'Restored Image')
 
-        st.download_button(
+        with col2:
+          st.image(output, f'Restored Image - {output}')
+          st.download_button(
             label="Download Image",
-            data=open(output, "rb").read(),
-            file_name="Image.jpg",
-        )
+            data=open(output, "rb"),
+            file_name=output
+          )
     else:
         st.warning('Upload a file', icon="⚠️")