Prefill annotated images

app.py

@@ -25,9 +25,10 @@ HEADER = f"""
 
 DEFAULT_TRANSFORM = "Rotate"
 
-DEFAULT_IMAGE = "images/doctor.webp"
-DEFAULT_IMAGE_HEIGHT = 400
-DEFAULT_IMAGE_WIDTH = 600
+DEFAULT_IMAGE_PATH = "images/doctor.webp"
+DEFAULT_IMAGE = np.array(Image.open(DEFAULT_IMAGE_PATH))
+DEFAULT_IMAGE_HEIGHT = DEFAULT_IMAGE.shape[0]
+DEFAULT_IMAGE_WIDTH = DEFAULT_IMAGE.shape[1]
 DEFAULT_BOXES = [
     [265, 121, 326, 177],  # Mask
     [192, 169, 401, 395],  # Coverall
@@ -144,7 +145,7 @@ def draw_not_implemented_image(image: np.ndarray, annotation_type: str):
     pil_image = Image.fromarray(image)
     draw = ImageDraw.Draw(pil_image)
     # align in the centerm, and make bigger font
-    text = f'Transform NOT working with "{annotation_type.upper()}" annotaions.'
+    text = f'Transform NOT working with "{annotation_type.upper()}" annotations.'
     length = draw.textlength(text)
     draw.text(
         (DEFAULT_IMAGE_WIDTH // 2 - length // 2, DEFAULT_IMAGE_HEIGHT // 2),
@@ -296,14 +297,19 @@ with gr.Blocks() as demo:
             )
             button = gr.Button("Apply!")
         image = gr.Image(
-            value=DEFAULT_IMAGE,
+            value=DEFAULT_IMAGE_PATH,
             type="numpy",
             height=500,
             width=300,
             sources=[],
         )
     with gr.Row():
-        augmented_image = gr.Gallery(rows=1, columns=3)
+        augmented_image = gr.Gallery(
+            value=update_augmented_images(DEFAULT_IMAGE, "A.NoOp()"),
+            rows=1,
+            columns=3,
+            show_label=False,
+        )
 
     select.change(fn=update_code_and_docs, inputs=[select], outputs=[code, docs])
     button.click(