Fixed some stuff

README.md

@@ -1,10 +1,10 @@
 ---
 title: Bounded Attention
-emoji: 🦀
+emoji: �
 colorFrom: pink
 colorTo: yellow
 sdk: gradio
-sdk_version: 4.22.0
+sdk_version: 3.43.2
 app_file: app.py
 pinned: false
 license: cc-by-nc-sa-3.0

app.py

@@ -13,6 +13,7 @@ from functools import partial
 
 RESOLUTION = 512
 MIN_SIZE = 0.01
+WHITE = 255
 COLORS = ["red", "blue", "green", "orange", "purple", "turquoise", "olive"]
 
 
@@ -80,10 +81,6 @@ def generate(
     seed,
     boxes
 ):
-    if 'boxes' not in boxes:
-        boxes['boxes'] = []
-
-    boxes = boxes['boxes']
     subject_token_indices = convert_token_indices(subject_token_indices, nested=True)
     if len(boxes) != len(subject_token_indices):
         raise ValueError("""
@@ -115,13 +112,11 @@ def convert_token_indices(token_indices, nested=False):
     return [int(index.strip()) for index in token_indices.split(',') if len(index.strip()) > 0]
 
 
-def draw(boxes, mask):
-    print('Called draw')
-    print('before boxes', boxes)
+def draw(boxes, mask, layout_image):
     if mask.ndim == 3:
-        mask = 255 - mask[..., 0]
+        mask = WHITE - mask[..., 0]
 
-    mask = (mask != 0).astype('uint8') * 255
+    mask = (mask != 0).astype('uint8') * WHITE
     if mask.sum() > 0:
         x1x2 = np.where(mask.max(0) != 0)[0] / RESOLUTION
         y1y2 = np.where(mask.max(1) != 0)[0] / RESOLUTION
@@ -130,9 +125,8 @@ def draw(boxes, mask):
 
         if (x2 - x1 > MIN_SIZE) and (y2 - y1 > MIN_SIZE):
             boxes.append((x1, y1, x2, y2))
-            layout_image = draw_boxes(np.array(boxes) * RESOLUTION)
+            layout_image = draw_boxes(boxes)
 
-    print('after boxes', boxes)
     return [boxes, None, layout_image]
 
 
@@ -140,10 +134,10 @@ def draw_boxes(boxes):
     if len(boxes) == 0:
         return None
 
-    image = Image.new('RGB', (RESOLUTION, RESOLUTION), (255, 255, 255))
+    boxes = np.array(boxes) * RESOLUTION
+    image = Image.new('RGB', (RESOLUTION, RESOLUTION), (WHITE, WHITE, WHITE))
     drawing = ImageDraw.Draw(image)
-    print(boxes)
-    for i, box in enumerate(boxes):
+    for i, box in enumerate(boxes.astype(int).tolist()):
         drawing.rectangle(box, outline=COLORS[i % len(COLORS)], width=4)
 
     return image
@@ -294,16 +288,9 @@ def main():
 
             boxes = gr.State([])
 
-            demo.load(
-                clear,
-                inputs=[batch_size],
-                outputs=[boxes, sketch_pad, layout_image, out_images],
-                queue=False
-            )
-
             sketch_pad.edit(
                 draw,
-                inputs=[boxes, sketch_pad],
+                inputs=[boxes, sketch_pad, layout_image],
                 outputs=[boxes, sketch_pad, layout_image],
                 queue=False,
             )
@@ -347,7 +334,7 @@ def main():
         description = """<p> The source code of this demo is based on the <a href="https://huggingface.co/spaces/gligen/demo/tree/main">GLIGEN demo</a>.</p>"""
         gr.HTML(description)
 
-    demo.queue(concurrency_count=1, api_open=False)
+    demo.queue(max_size=50)
     demo.launch(share=False, show_api=False, show_error=True)
 
 if __name__ == '__main__':