changed the way the useModel flag is handled

Signed-off-by: Blessing Osakue <iyoworks@gmail.com>