Update main.py

main.py

@@ -97,11 +97,9 @@ 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)
+        upscaled_image_bytes = upscale_image_stable_diffusion(resized_image_bytes) # Change this line
         
         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)