Update train.py

train.py

@@ -82,6 +82,7 @@ def to_io(example):
     if isinstance(out, dict):
         target = json.dumps(out, separators=(",", ":"))
     else:
+            # extract JSON from text
         m = re.search(r"\{.*\}", out, re.S)
         target = m.group(0) if m else out
         target = json.dumps(json.loads(target), separators=(",", ":"))