Improve model search model function

Improve model search model function