import gradio as gr | |
from PIL import Image | |
def add_watermark(image): | |
image = Image.fromarray(np.uint8(image)) | |
watermark = Image.open("watermark.png") | |
image = image.convert("RGBA") | |
watermark = watermark.convert("RGBA") | |
width = image.width() | |
height = image.height() | |
image.paste(watermark, (width, height), watermark) | |
return image | |
gr.Interface(fn=add_watermark, | |
inputs="image", | |
outputs="image", | |
title="Add watermark", | |
examples = ["example.jpg"] | |
).launch(); |