blanchon commited on
Commit
83a5122
·
1 Parent(s): f1ad758

Remove tonemapping from kwargs for ldr images

Browse files
Files changed (1) hide show
  1. x2rgb/gradio_demo_x2rgb.py +3 -0
x2rgb/gradio_demo_x2rgb.py CHANGED
@@ -50,6 +50,9 @@ def generate(
50
  if file.name.endswith(".exr"):
51
  return load_exr_image(file.name, **kwargs).to("cuda")
52
  elif file.name.endswith((".png", ".jpg", ".jpeg")):
 
 
 
53
  return load_ldr_image(file.name, **kwargs).to("cuda")
54
  return None
55
 
 
50
  if file.name.endswith(".exr"):
51
  return load_exr_image(file.name, **kwargs).to("cuda")
52
  elif file.name.endswith((".png", ".jpg", ".jpeg")):
53
+ # remove tonemapping from kwargs for ldr images
54
+ if "tonemaping" in kwargs:
55
+ kwargs.pop("tonemaping")
56
  return load_ldr_image(file.name, **kwargs).to("cuda")
57
  return None
58