add application

.python-version

@@ -0,0 +1 @@
+3.10.16

README.md

@@ -1,13 +1,18 @@
 ---
 title: Finegrain Light Switcher
-emoji: ⚡
-colorFrom: red
-colorTo: yellow
+emoji: 💡
+colorFrom: yellow
+colorTo: indigo
 sdk: gradio
-sdk_version: 5.22.0
+sdk_version: 4.44.1
+python_version: 3.10
 app_file: app.py
 pinned: false
+license: other
 short_description: Switch lamps on in your images.
+tags:
+- refiners
+- image-to-image
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# Finegrain Light Switcher

app.py

@@ -0,0 +1,116 @@
+import os
+import random
+import time
+from typing import Any
+
+import gradio as gr
+import pillow_avif  # noqa: F401
+import pillow_heif
+import spaces
+import torch
+from gradio_imageslider import ImageSlider
+from PIL import Image
+
+
+class ModuleInterface:
+    def __init__(self, d: dict[str, Any]):
+        self.d = d
+
+    @property
+    def Model(self) -> Any:
+        return self.d["Model"]
+
+    def process(self, input_image: Image.Image, model: Any, seed: int) -> Image.Image:
+        return self.d["process"](input_image, model, seed)
+
+
+assert (src := os.getenv("LIGHT_SWITCHER_LITE")), "LIGHT_SWITCHER_LITE not set"
+exec_globals: dict[str, Any] = {}
+exec(src, exec_globals)
+light_switcher_lite = ModuleInterface(exec_globals)
+
+pillow_heif.register_avif_opener()
+
+
+DEVICE_CPU = torch.device("cpu")
+DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+DTYPE = torch.bfloat16 if torch.cuda.is_bf16_supported() else torch.float32
+
+# CPU -> GPU dance because of ZeroGPU
+
+path = "finegrain/weights-light-switcher-space"
+model = light_switcher_lite.Model.from_pretrained(path, device=DEVICE_CPU, dtype=DTYPE)
+
+model.to(DEVICE)
+
+
+@spaces.GPU
+def process(input_image: Image.Image, seed: int = 42) -> tuple[tuple[Image.Image, Image.Image], dict[str, Any]]:
+    output_image = light_switcher_lite.process(input_image, model, seed)
+    resized_input_image = input_image.resize(output_image.size)
+    return ((resized_input_image, output_image), gr.update(value=random.choice(BUTTON_LABELS)))
+
+
+TITLE = """
+<h1>Finegrain Light Switcher</h1>
+<p>
+  Given an image with a lamp switched off, the model should turn it on.
+</p>
+<p>
+  🔌 For higher resolution results with control over lighting intensity and warmth,&nbsp;
+  <a href="https://finegrain.ai">head to the Finegrain API</a> 🔌
+</p>
+<p>
+  <a href="https://discord.gg/zFKg5TjXub" target="_blank">[Discord]</a>
+  <a href="https://github.com/finegrain-ai" target="_blank">[GitHub]</a>
+  <a href="https://finegrain.ai">[Finegrain API]</a>
+</p>
+"""
+
+BUTTON_LABELS = [
+    "💡",
+    "Let there be light!",
+    "Light it up like a Christmas tree!",
+    "Turn it on!",
+    "Aziz, Light!",
+    "Make it shine ✨",
+    "Flip the magic switch.",
+]
+
+random.seed(time.time())
+
+with gr.Blocks() as demo:
+    gr.HTML(TITLE)
+    with gr.Row():
+        with gr.Column():
+            input_image = gr.Image(type="pil", label="Input Image")
+            run_button = gr.ClearButton(components=None, value=random.choice(BUTTON_LABELS))
+        with gr.Column():
+            output_slider = ImageSlider(label="Before / After")
+            run_button.add(output_slider)
+
+    with gr.Accordion("Advanced Options", open=False):
+        seed = gr.Slider(minimum=0, maximum=999, value=42, step=1, label="Seed")
+
+    run_button.click(
+        fn=process,
+        inputs=[input_image, seed],
+        outputs=[output_slider, run_button],
+    )
+
+    gr.Examples(
+        examples=[
+            "examples/01.webp",
+            "examples/02.webp",
+            "examples/03.webp",
+            "examples/04.webp",
+            "examples/05.webp",
+        ],
+        inputs=[input_image],
+        outputs=[output_slider, run_button],
+        fn=process,
+        cache_examples=True,
+        run_on_click=False,
+    )
+
+demo.launch(share=False)

pyproject.toml

@@ -0,0 +1,27 @@
+[project]
+name = "light-switcher"
+version = "0.1.0"
+description = "Finegrain Light Switcher (Gradio)"
+authors = [
+    { name = "Pierre Chapuis", email = "[email protected]" }
+]
+
+[tool.ruff]
+line-length = 120
+target-version = "py310"
+
+[tool.ruff.lint]
+select = [
+  "E",   # pycodestyle errors
+  "W",   # pycodestyle warnings
+  "F",   # pyflakes
+  "UP",  # pyupgrade
+  "A",   # flake8-builtins
+  "B",   # flake8-bugbear
+  "Q",   # flake8-quotes
+  "I",   # isort
+]
+
+[tool.pyright]
+include = ["app.py"]
+exclude = ["**/__pycache__"]

requirements.txt

@@ -0,0 +1,9 @@
+git+https://github.com/huggingface/diffusers@97abdd2210a540c2e71aee63c80a22723031cd57
+git+https://github.com/finegrain-ai/refiners@d288e94fa8eed1386bd28cd0d5ceb8109c3ff398
+gradio_imageslider==0.0.20
+spaces==0.32.0
+numpy<2.0.0
+pillow>=10.4.0
+pillow-heif>=0.18.0
+accelerate>=1.3.0
+pillow-avif-plugin>=1.5.0