Avoid coord error with lacking ":".

app.py

@@ -90,7 +90,11 @@ def prompt_to_layout(user_prompt, intensity, fpath=None):
  output = tokenizer.batch_decode(output, skip_special_tokens=True)
  layout = output[0].split('[User prompt]')[1].split('[Layout] ')[1].split(', ')
  spaces = [txt.split(':')[0] for txt in layout]
- coords = [txt.split(':')[1].rstrip() for txt in layout]
+ coords = []
+ for txt in layout:
+ if ':' in txt:
+ split_txt = txt.split(':')
+ coords.append(split_txt[1].rstrip())
  coordinates = [re.findall(regex, coord) for coord in coords]
  
  num_coords = []