Short interview with HR, then technical interview with Team Lead. Technical interview was an open discussion followed by a real world problem to fix using ML. They are looking for a problem solver, so you should discuss all possible solutions that come in your mind.