watermark_demo / app.py
meg's picture
meg HF staff
Update app.py
b15d92f verified
raw
history blame
1.49 kB
import gradio as gr
import os
a = os.path.join(os.path.dirname(__file__), "files/barkley_balloon.mp4")
b = os.path.join(os.path.dirname(__file__), "files/convert-to-gradio-sd.mp4")
w1 = os.path.join(os.path.dirname(__file__), "files/AI_generated.png")
w2 = os.path.join(os.path.dirname(__file__), "files/hf-logo.png")
w3 = os.path.join(os.path.dirname(__file__), "files/watermark_qr_code.png")
def generate_video(original_video, watermark):
return gr.Video(original_video, watermark=watermark)
# Use RGBA image mode to preserve transparency for png images.
with gr.Blocks() as demo:
gr.Markdown("# Watermarking with Gradio: Example")
gr.Markdown("Watermarks can be **visible** or **invisible**.")
gr.Markdown("""They can provide information directly, or provide a link for more information.
Visible watermarks can help to disclose when content is AI-generated, so viewers know it's synthetic.
Invisible watermarks can help to mark content as authentic.""")
gr.Markdown()
gr.Markdown("""For more information on watermarking -- what watermarking is, why it's important, and the tools available on Hugging Face -- please check out [our blogpost on AI watermarking](https://huggingface.co/blog/watermarking).""")
gr.Interface(generate_video, [gr.Video(), gr.Image(type='filepath', image_mode="RGBA")], gr.Video(),
examples=[[a, w1], [b, w2], [a, w3]])
if __name__ == "__main__":
demo.launch()