Removed argument classFileName to only use the model name