Removed examples

app.py

@@ -18,30 +18,6 @@ MIN_SIZE = 0.01
 WHITE = 255
 COLORS = ["red", "blue", "green", "orange", "purple", "turquoise", "olive"]
 
-PROMPT1 = "3D Pixar animation of a cute unicorn and a pink hedgehog and a nerdy owl traveling in a magical forest"
-PROMPT2 = "science fiction movie poster with an astronaut and a robot and a green alien and a spaceship"
-PROMPT3 = "a golden retriever and a german shepherd and a boston terrier and an english bulldog and a border collie in a pool"
-EXAMPLE_BOXES = {
-    PROMPT1 : [
-        [0.35, 0.4, 0.65, 0.9],
-        [0, 0.6, 0.3, 0.9],
-        [0.7, 0.55, 1, 0.85]
-    ],
-    PROMPT2: [
-        [0.4, 0.45, 0.6, 0.95],
-        [0.2, 0.3, 0.4, 0.85],
-        [0.6, 0.3, 0.8, 0.85],
-        [0.1, 0, 0.9, 0.3]
-    ],
-    PROMPT3: [
-        [0, 0.5, 0.2, 0.8],
-        [0.2, 0.2, 0.4, 0.5],
-        [0.4, 0.5, 0.6, 0.8],
-        [0.6, 0.2, 0.8, 0.5],
-        [0.8, 0.5, 1, 0.8]
-    ],
-}
-
 DESCRIPTION = """
     <p style="text-align: center; font-weight: bold;">
     <span style="font-size: 28px">Bounded Attention</span>
@@ -203,13 +179,6 @@ def clear(batch_size):
     return [[], None, None, None]
 
 
-def make_example_inputs(prompt):
-    boxes = EXAMPLE_BOXES[prompt]
-    sketchpad = draw_boxes(boxes, is_sketch=True)
-    layout_image = draw_boxes(boxes)
-    return sketchpad, layout_image, prompt
-
-
 def main():
     css = """
     #paper-info a {
@@ -278,8 +247,8 @@ def main():
             )
 
             with gr.Row():
-                sketchpad = gr.Sketchpad(label="Sketch Pad (draw each bounding box in a different layer)", width=RESOLUTION, height=RESOLUTION)
-                layout_image = gr.Image(type="pil", label="Bounding Boxes", interactive=False, width=RESOLUTION, height=RESOLUTION, scale=1)
+                sketchpad = gr.Sketchpad(label="Sketch Pad (draw each bounding box in a different layer)")
+                layout_image = gr.Image(type="pil", label="Bounding Boxes", interactive=False)
 
             with gr.Row():
                 clear_button = gr.Button(value="Clear")
@@ -367,37 +336,8 @@ def main():
 
         with gr.Column():
             gr.Examples(
-                examples=[
-                    [
-                        *make_example_inputs(PROMPT1),
-                        "7,8,17;11,12,17;15,16,17", "5,6,9,10,13,14,18,19", "21",
-                        25, 10, 3, 1, 1,
-                        7.5, 1, 5, 0.2, 15,
-                        286,
-                    ],
-                    [
-                        *make_example_inputs(PROMPT2),
-                        "7;10;13,14;17", "5,6,8,9,11,12,15,16", "17",
-                        18, 5, 3, 1, 1,
-                        7.5, 1, 5, 0.2, 15,
-                        216,
-                    ],
-                    [
-                        *make_example_inputs(PROMPT3),
-                        "2,3;6,7;10,11;14,15;18,19", "1,4,5,8,9,12,13,16,17,20,21", "22",
-                        18, 5, 3, 1, 1,
-                        7.5, 1, 5, 0.2, 15,
-                        156,
-                    ],
-                ],
-                inputs=[
-                    sketchpad, layout_image,
-                    prompt, subject_token_indices, filter_token_indices, num_tokens,
-                    init_step_size, final_step_size, num_clusters_per_subject, cross_loss_scale, self_loss_scale,
-                    classifier_free_guidance_scale, batch_size, num_iterations, loss_threshold, num_guidance_steps,
-                    seed,
-                ],
-                outputs=[boxes, sketchpad, layout_image, out_images],
+                examples=[["a ginger kitten and a gray puppy in a yard", "2,3;6,7", "1,4,5,8,9", "10"]],
+                inputs=[prompt, subject_token_indices, filter_token_indices, num_tokens],
             )
 
         gr.HTML(FOOTNOTE)