app.py

import gradio as gr
import numpy as np
import requests
import imageio
from PIL import Image
from io import BytesIO

def dummy(img):
    imageio.imwrite("output_image.png", img["mask"])
    return img["image"], img["mask"]

# 图片的 URL 地址
image_url = "https://api.telegram.org/file/bot6584991592:AAGNv881SaUQd9zmDs2OzQEpKD7nrMkzMpk/photos/file_3648.jpg"

# 使用 requests 获取图片数据
response = requests.get(image_url)
image = Image.open(BytesIO(response.content))
default_image = np.array(image)

with gr.Blocks() as demo:
    with gr.Row():
        # 使用从URL下载的图片数组初始化Image组件
        img = gr.Image(tool="sketch", label="base image", show_label=True, value=default_image)
    with gr.Row():
        img1 = gr.Image()
        img2 = gr.Image(label="mask image", show_label=True)
    btn = gr.Button()
    btn.click(dummy, img, [img1, img2])

demo.launch(debug=True)