import gradio as gr
import numpy as np
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")

    watermark = watermark.resize((image.width, image.height))
    image.paste(watermark,(0,0),watermark)

    return image

gr.Interface(fn=add_watermark,
             inputs="image",
             outputs="image",
             title="Add watermark" 
            ).launch();