Spaces:

theSure
/

Omnieraser

Running on Zero

theSure commited on 16 days ago

Commit

56c7591

verified ·

1 Parent(s): b1e0393

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -101,7 +101,7 @@ def predict(
         srgb_profile = ImageCms.createProfile("sRGB")
         io_handle = io.BytesIO(icc_profile)
         src_profile = ImageCms.ImageCmsProfile(io_handle)
-        input_image = ImageCms.profileToProfile(input_image["background"], src_profile, srgb_profile)
         input_image.info.pop('icc_profile', None)
     if size1 < size2:

         srgb_profile = ImageCms.createProfile("sRGB")
         io_handle = io.BytesIO(icc_profile)
         src_profile = ImageCms.ImageCmsProfile(io_handle)
+        input_image = ImageCms.profileToProfile(input_image, src_profile, srgb_profile)
         input_image.info.pop('icc_profile', None)
     if size1 < size2: