app.py

import gradio as gr
import lightning as L

from src.gradio_demo.color_matching import create_color_matching
from src.gradio_demo.sw_guidance import create_sw_guidance

with gr.Blocks() as demo:
    gr.Markdown(
        """
        # ReSWD

        ReSTIR‘d, not shaken. Combining Reservoir Sampling and Sliced Wasserstein
        Distance for Variance Reduction.

        ReSWD is a method for distribution matching with reduced variance.
        """
    )
    fabric = L.Fabric(devices=1, accelerator="auto", precision="16-mixed")

    with gr.Tab("SW Guidance (SD 3.5 Large Turbo)"):
        create_sw_guidance(fabric, "stabilityai/stable-diffusion-3.5-large-turbo")
    with gr.Tab("Color Matching"):
        create_color_matching(fabric)

    demo.launch()