delete redundancy line `if args.train`, simplify code.
Attach a file by drag & drop or click to upload
