Add inference module