Generalizable Human-Robot Collaborative Assembly Using Imitation Learning and Force Control