Update pages/04_Super_Resolution.py

pages/04_Super_Resolution.py

@@ -219,23 +219,22 @@ outscale = st.slider("Image Upscaling Factor", 1, 10, 2)
 face_enhance = st.checkbox("Face Enhancement using GFPGAN (Doesn't work for anime images)")
 
 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:
-    col1.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')
@@ -246,14 +245,12 @@ if restore:
     if input_img is not None:
         output = realesrgan(input_img, model_name, denoise_strength, 
                             face_enhance, outscale)
-        #im = Image.fromarray(output)
-        #im.save("Image.jpg")
-        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="⚠️")