Update main.py

main.py

@@ -239,6 +239,7 @@ def init_shapes(num_paths,
                 shape_cnt,
                 pos_init_method=None,
                 trainable_stroke=False,
+                gt=None,
                 **kwargs):
     shapes = []
     shape_groups = []
@@ -289,7 +290,7 @@ def init_shapes(num_paths,
         shapes.append(path)
         # !!!!!!problem is here. the shape group shape_ids is wrong
 
-        if 'gt' in kwargs:
+        if gt is not None:
             wref, href = color_ref
             wref = max(0, min(int(wref), w-1))
             href = max(0, min(int(href), h-1))