Update app.py

app.py

@@ -88,7 +88,7 @@ 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] for txt in layout]
+    coords = [txt.split(':')[1].rstrip() for txt in layout]
     coordinates = [re.findall(regex, coord) for coord in coords]
     layout_dict = dict(zip(spaces, coords))
     polygons = []