AI IN THE DESIGN PROCESS: TRAINING THE HUMAN-AI COLLABORATION