Improve API definition

• remove misleading headers from the UI
• add training_args, training_output and training_duration to the training_response

@alvarolopez if you are fine with the changes you can close #74 (closed)