app.py

import gradio as gr
from PIL import Image

import sys
import os

# Download code from private model repo
from huggingface_hub import snapshot_download

# # Download your private code
# repo_path = snapshot_download(
#     repo_id="limitedonly41/bg_remove",
#     repo_type="model",
#     token=os.getenv("HF_TOKEN"),  # Will use Space's token automatically
# )


ASSET_DIR = snapshot_download("limitedonly41/bg_remove", token=os.getenv("HF_TOKEN"), repo_type="model")

EXAMPLE_PERSON = os.path.join(ASSET_DIR, "examples/person.jpeg")
EXAMPLE_BACKGROUND = os.path.join(ASSET_DIR, "examples/forest.jpeg")

sys.path.insert(0, ASSET_DIR)

from bg_remove import replace_background


def process(person_img, bg_img):
    return replace_background(person_img, bg_img)

with gr.Blocks() as demo:
    gr.Markdown("## Background Replacement (Instant Startup, ZeroGPU)")

    with gr.Row():
        person = gr.Image(type="pil", label="Person", value=EXAMPLE_PERSON)
        background = gr.Image(type="pil", label="Background", value=EXAMPLE_BACKGROUND)

    out = gr.Image(label="Result")
    btn = gr.Button("Replace Background")

    btn.click(process, inputs=[person, background], outputs=out)

demo.launch()


# Add to Python path
sys.path.insert(0, repo_path)

from bg_remove import replace_background   # hidden code

def process(person_img, bg_img):
    result = replace_background(person_img, bg_img)
    return result

with gr.Blocks() as demo:
    gr.Markdown("## Background Replacement (Demo @Tair)")

    with gr.Row():
        person = gr.Image(type="pil", label="Upload person photo")
        background = gr.Image(type="pil", label="Upload background")

    output = gr.Image(label="Output")

    run_btn = gr.Button("Replace Background")
    run_btn.click(process, inputs=[person, background], outputs=output)

demo.launch()