tiny issue in model_builder_hook.py

[IIAT-MR-LL]

There is compile issue in model_builder_hook.py, but it still can get the result base on readme.

it might be the error is exclude by "if else", so it no big impact if follow the readme