Update pages/04_Super_Resolution.py

pages/04_Super_Resolution.py

@@ -219,22 +219,23 @@ 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
-  cols = st.columns(2)
+  col1, col2 = st.columns(2)
 
-  cols[0].image(input_img, 'Source Image')
+  with col1:
+    col1.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')
@@ -245,10 +246,12 @@ 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')
-        output_img.download_button(
+        output.save("Image.jpg")
+        with col2:
+          st.image(output, 'Restored Image')
+          st.download_button(
             label="Download Image",
-            data=open(output, "rb").read(),
+            data=open("Image.jpg", "rb"),
             file_name="Image.jpg",
         )
     else: