mirror of
https://github.com/gradio-app/gradio.git
synced 2024-12-21 02:19:59 +08:00
9b42ba8f10
* changes * changes * revert changes * changes * add changeset * notebooks script * changes * changes --------- Co-authored-by: Ali Abid <aliabid94@gmail.com> Co-authored-by: gradio-pr-bot <gradio-pr-bot@users.noreply.github.com> Co-authored-by: Ali Abdalla <ali.si3luwa@gmail.com>
31 lines
764 B
Python
31 lines
764 B
Python
import gradio as gr
|
|
import pathlib
|
|
from PIL import Image
|
|
import numpy as np
|
|
import urllib.request
|
|
|
|
source_dir = pathlib.Path(__file__).parent
|
|
|
|
urllib.request.urlretrieve(
|
|
'https://gradio-builds.s3.amazonaws.com/demo-files/base.png',
|
|
str(source_dir / "base.png")
|
|
)
|
|
urllib.request.urlretrieve(
|
|
"https://gradio-builds.s3.amazonaws.com/demo-files/buildings.png",
|
|
str(source_dir / "buildings.png")
|
|
)
|
|
|
|
base_image = Image.open(str(source_dir / "base.png"))
|
|
building_image = Image.open(str(source_dir / "buildings.png"))
|
|
|
|
# Create segmentation mask
|
|
building_image = np.asarray(building_image)[:, :, -1] > 0
|
|
|
|
with gr.Blocks() as demo:
|
|
gr.AnnotatedImage(
|
|
value=(base_image, [(building_image, "buildings")]),
|
|
height=500,
|
|
)
|
|
|
|
demo.launch()
|