Remove tonemapping from kwargs for ldr images
Browse files
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 |
|