Update agent.py according to changes noted https://github.com/tensorflow/models/issues/7719
Attach a file by drag & drop or click to upload
