fix calculating model size

[masoud-khalilian]

Sometimes when models attempt to publish the result of model size, it will through an error. Check the cause and resolve it.