Update pages/05_Image_Denoizer.py

pages/05_Image_Denoizer.py

@@ -214,23 +214,22 @@ outscale = 1
 face_enhance = False
 
 if input_img:
-  #print(input_img)
+  print(input_img)
   input_img = Image.open(input_img)
   # Display image properties
-  col1, col2 = st.columns(2)
+  cols = st.columns(2)
 
-  with col1:
-    st.image(input_img, 'Source Image')
+  cols[0].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')
@@ -241,13 +240,13 @@ if restore:
     if input_img is not None:
         output = realesrgan(input_img, model_name, denoise_strength, 
                             face_enhance, outscale)
-        with col2:
-          st.image(output, f'Restored Image - {output}')
-          st.download_button(
+        output_img.image(output, 'Restored Image')
+        
+        st.download_button(
             label="Download Image",
-            data=open(output, "rb"),
-            file_name=output
-          )
+            data=open(output, "rb").read(),
+            file_name="Image.jpg"
+        )
     else:
         st.warning('Upload a file', icon="⚠️")