app.py

import gradio as gr

from chrislib.general import uninvert, view

from intrinsic.pipeline import run_pipeline
from intrinsic.model_util import load_models

global intrinsic_model
intrinsic_model = load_models('paper_weights', device='cpu')

def decompose(img):
    result = run_pipeline(
        img,
        intrinsic_model,
        device='cpu'
    )

    return view(uninvert(result['inv_shd'])), "Completed"


gr.Interface(
    decompose,
    [gr.inputs.Image(type="numpy", label="Input")],
    [gr.outputs.Image(type="numpy", label="Output"), gr.outputs.Textbox(label=":")],
    examples=[["avocado.jpg"]]
).launch(debug=True)