Update main.py

main.py

@@ -97,9 +97,11 @@ def main():
         resized_image_bytes = resize_image(image_bytes)
         
         st.success("Upscaling image with stable-diffusion-x4-latent-upscaler...")
+        # use the resized_image_bytes instead of the original image_bytes
         upscaled_image_bytes = upscale_image_stable_diffusion(resized_image_bytes)
         
         st.success("Further upscaling image with GFPGAN...")
+        # Again use the upscaled_image_bytes from the previous step for the GFPGAN
         img = further_upscale_image(upscaled_image_bytes)
         
         st.image(img, caption='Upscaled Image', use_column_width=True)