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

        <a href="https://reservoirswd.github.io/"><img src="https://img.shields.io/badge/Project%20Page-5CE1BC.svg"></a>
        <a href="https://reservoirswd.github.io/static/paper.pdf"><img src="https://img.shields.io/badge/Arxiv-2408.00653-B31B1B.svg"></a>
        <a href="https://github.com/Stability-AI/ReSWD"><img src="https://img.shields.io/badge/git-%23F05032?logo=git&logoColor=white"></a>
        <br>

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

        ReSWD is a method for distribution matching with reduced variance.
        """
    )
    with gr.Tab("SW Guidance (SD 3.5 Large Turbo)"):
        create_sw_guidance("stabilityai/stable-diffusion-3.5-large-turbo")
    with gr.Tab("Color Matching"):
        create_color_matching()

    demo.launch()